fix travis tests

openedx-unsupported · May 22, 2018 · 07343a0 · 07343a0
1 parent bba3306
commit 07343a0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 9 deletions.
diff --git a/edx/analytics/tasks/insights/tests/test_user_activity.py b/edx/analytics/tasks/insights/tests/test_user_activity.py
@@ -6,16 +6,14 @@
 import json
 from unittest import TestCase
 
-from ddt import data, ddt, unpack
 from luigi import date_interval
-from mock import Mock, call
 
+from ddt import data, ddt, unpack
 from edx.analytics.tasks.common.tests.map_reduce_mixins import MapperTestMixin, ReducerTestMixin
-from edx.analytics.tasks.insights.user_activity import (
-    InsertToMysqlCourseActivityTask, UserActivityTask
-)
+from edx.analytics.tasks.insights.user_activity import InsertToMysqlCourseActivityTask, UserActivityTask
 from edx.analytics.tasks.util.constants import PredicateLabels
 from edx.analytics.tasks.util.tests.opaque_key_mixins import InitializeLegacyKeysMixin, InitializeOpaqueKeysMixin
+from mock import Mock, call
 
 
 @ddt

diff --git a/edx/analytics/tasks/monitor/overall_events.py b/edx/analytics/tasks/monitor/overall_events.py
@@ -53,4 +53,4 @@ def spark_job(self, *args):
         df = self.get_event_log_dataframe(self._spark)
         df = df.groupBy('event_date').count()
         df.coalesce(1).write.csv(self.output().path, mode='overwrite', sep='\t')
-        # df.repartition(1).rdd.map(lambda row: '\t'.join(map(str, row))).saveAsTextFile(self.output_dir().path)
+        # df.repartition(1).rdd.map(lambda row: '\t'.join(map(str, row))).saveAsTextFile(self.output_dir().path)
diff --git a/edx/analytics/tasks/monitor/total_events_report.py b/edx/analytics/tasks/monitor/total_events_report.py
@@ -7,10 +7,9 @@
 
 from edx.analytics.tasks.common.mapreduce import MapReduceJobTaskMixin
 from edx.analytics.tasks.common.pathutil import EventLogSelectionDownstreamMixin
-from edx.analytics.tasks.monitor.overall_events import TotalEventsDailyTask, SparkTotalEventsDailyTask
-from edx.analytics.tasks.util.url import ExternalURL, get_target_from_url
+from edx.analytics.tasks.monitor.overall_events import SparkTotalEventsDailyTask, TotalEventsDailyTask
 from edx.analytics.tasks.util.overwrite import OverwriteOutputMixin
-
+from edx.analytics.tasks.util.url import ExternalURL, get_target_from_url
 
 log = logging.getLogger(__name__)