Fix imports.

bonobo/execution/__init__.py

@@ -1,3 +1,5 @@
 import logging
 
 logger = logging.getLogger(__name__)
+
+__all__ = []

bonobo/execution/base.py

@@ -1,17 +1,14 @@
-import logging
+import sys
 from contextlib import contextmanager
 from logging import WARNING, ERROR
 from time import sleep
 
-import sys
-
 import mondrian
-
 from bonobo.config import create_container
 from bonobo.config.processors import ContextCurrifier
+from bonobo.execution import logger
 from bonobo.util import isconfigurabletype
 from bonobo.util.objects import Wrapper, get_name
-from bonobo.execution import logger
 
 
 @contextmanager

bonobo/strategies/executor.py

@@ -73,7 +73,9 @@ class ExecutorStrategy(Strategy):
                     try:
                         plugin.loop()
                     except Exception:
-                        mondrian.excepthook(*sys.exc_info(), context='In plugin loop for {}...'.format(get_name(plugin)))
+                        mondrian.excepthook(
+                            *sys.exc_info(), context='In plugin loop for {}...'.format(get_name(plugin))
+                        )
 
             futures.append(executor.submit(_runner))
 

bonobo/util/testing.py

@@ -1,7 +1,7 @@
 from contextlib import contextmanager
 
 from bonobo import open_fs, Token
-from bonobo.execution import GraphExecutionContext
+from bonobo.execution.graph import GraphExecutionContext
 from bonobo.execution.node import NodeExecutionContext
 
 

tests/io/test_csv.py

@@ -1,7 +1,6 @@
 import pytest
 
-from bonobo import Bag, CsvReader, CsvWriter, settings
+from bonobo import CsvReader, CsvWriter, settings
-from bonobo.constants import BEGIN, END
 from bonobo.execution.node import NodeExecutionContext
 from bonobo.util.testing import FilesystemTester, BufferingNodeExecutionContext
 

tests/test_basicusage.py

@@ -1,8 +1,9 @@
+from unittest.mock import patch
+
 import pytest
 
 import bonobo
-from bonobo.execution import GraphExecutionContext
+from bonobo.execution.graph import GraphExecutionContext
-from unittest.mock import patch
 
 
 @pytest.mark.timeout(2)