flake

force_bdss/io/tests/test_workflow_writer.py

@@ -28,7 +28,7 @@ class TestWorkflowWriter(unittest.TestCase):
             bundle=mock_mco_bundle
         )
         mock_mco_bundle.create_model = mock.Mock(
-            return_value = mock_mco_model
+            return_value=mock_mco_model
         )
         self.mock_registry.mco_bundle_by_id = mock.Mock(
             return_value=mock_mco_bundle)

force_bdss/io/workflow_writer.py

 import json
-from traits.api import Instance, HasStrictTraits
-
-from ..bundle_registry_plugin import BundleRegistryPlugin
+from traits.api import HasStrictTraits
 
 
 class WorkflowWriter(HasStrictTraits):