Adjust some import order.

PiperOrigin-RevId: 252134806
tensorflow · Jun 7, 2019 · 7b608dc · 7b608dc
1 parent ae6cbfa
commit 7b608dc
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 6 deletions.
diff --git a/tfx/examples/chicago_taxi_pipeline/taxi_pipeline_simple_airflow_e2e_test.py b/tfx/examples/chicago_taxi_pipeline/taxi_pipeline_simple_airflow_e2e_test.py
@@ -20,11 +20,13 @@
 import subprocess
 import tempfile
 import time
-import unittest
 import pytest
 
 import tensorflow as tf
 from typing import Sequence, Set, Text
+
+import unittest
+
 from tfx.utils import io_utils
 
 
@@ -99,7 +101,7 @@ def _CheckPendingTasks(self, pending_task_names: Sequence[Text]) -> Set[Text]:
     for task, state in failed.items():
       tf.logging.error('Retrieving logs for %s task %s', state, task)
       self._PrintTaskLogsOnError(task)
-    self.assertEqual(0, len(failed))
+    self.assertFalse(failed)
     return still_pending
 
   def setUp(self):

diff --git a/tfx/orchestration/airflow/airflow_component_test.py b/tfx/orchestration/airflow/airflow_component_test.py
@@ -68,7 +68,9 @@ def setUp(self):
   @mock.patch('airflow.operators.python_operator.BranchPythonOperator')
   @mock.patch('airflow.operators.python_operator.PythonOperator')
   @mock.patch('airflow.operators.dummy_operator.DummyOperator')
-  @mock.patch('tfx.orchestration.airflow.airflow_adapter.AirflowAdapter')
+  @mock.patch(
+      'tfx.orchestration.airflow.airflow_adapter.AirflowAdapter'
+  )
   def test_tfx_workflow_non_docker(
       self, mock_airflow_adapter_class, mock_dummy_operator_class,
       mock_python_operator_class, mock_branch_python_operator_class):
@@ -153,7 +155,9 @@ def test_tfx_workflow_non_docker(
   @mock.patch('airflow.operators.python_operator.BranchPythonOperator')
   @mock.patch('airflow.operators.python_operator.PythonOperator')
   @mock.patch('airflow.operators.dummy_operator.DummyOperator')
-  @mock.patch('tfx.orchestration.airflow.airflow_adapter.AirflowAdapter')
+  @mock.patch(
+      'tfx.orchestration.airflow.airflow_adapter.AirflowAdapter'
+  )
   def test_tfx_workflow_docker(
       self, mock_airflow_adapter_class, mock_dummy_operator_class,
       mock_python_operator_class, mock_branch_python_operator_class):
@@ -219,7 +223,9 @@ def test_tfx_workflow_docker(
 
   @mock.patch('airflow.operators.python_operator.BranchPythonOperator')
   @mock.patch('airflow.operators.python_operator.PythonOperator')
-  @mock.patch('tfx.orchestration.airflow.airflow_adapter.AirflowAdapter')
+  @mock.patch(
+      'tfx.orchestration.airflow.airflow_adapter.AirflowAdapter'
+  )
   def test_airflow_component(self, mock_airflow_adapter_class,
                              mock_python_operator_class,
                              mock_branch_python_operator_class):

diff --git a/tfx/orchestration/kubeflow/container_image_e2e_test.py b/tfx/orchestration/kubeflow/container_image_e2e_test.py
@@ -18,12 +18,13 @@
 
 import os
 import tempfile
-import unittest
 
 import docker
 import pytest
 import tensorflow as tf
 
+import unittest
+
 
 @pytest.mark.end_to_end
 class ContainerImageEndToEndTest(unittest.TestCase):