From 1971d67ad5766110118a55de2457bf955541280a Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Thu, 20 Jul 2017 11:58:17 +0100
Subject: [PATCH] flake
---
force_bdss/io/tests/test_workflow_writer.py | 2 +-
force_bdss/io/workflow_writer.py | 4 +---
2 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/force_bdss/io/tests/test_workflow_writer.py b/force_bdss/io/tests/test_workflow_writer.py
index af25d56..e6f81ec 100644
--- a/force_bdss/io/tests/test_workflow_writer.py
+++ b/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)
diff --git a/force_bdss/io/workflow_writer.py b/force_bdss/io/workflow_writer.py
index d0b2c8d..129e647 100644
--- a/force_bdss/io/workflow_writer.py
+++ b/force_bdss/io/workflow_writer.py
@@ -1,7 +1,5 @@
import json
-from traits.api import Instance, HasStrictTraits
-
-from ..bundle_registry_plugin import BundleRegistryPlugin
+from traits.api import HasStrictTraits
class WorkflowWriter(HasStrictTraits):
--
GitLab