Add missing imports; enamed pipeline test module

golemfactory · Jul 25, 2023 · 7a2457e · 7a2457e
1 parent 951a496
commit 7a2457e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/golem/managers/activity/__init__.py b/golem/managers/activity/__init__.py
@@ -1,10 +1,12 @@
 from golem.managers.activity.defaults import default_on_activity_start, default_on_activity_stop
+from golem.managers.activity.mixins import ActivityPrepareReleaseMixin
 from golem.managers.activity.pool import ActivityPoolManager
 from golem.managers.activity.single_use import SingleUseActivityManager
 
 __all__ = (
-    "ActivityPoolManager",
-    "SingleUseActivityManager",
     "default_on_activity_start",
     "default_on_activity_stop",
+    "ActivityPrepareReleaseMixin",
+    "ActivityPoolManager",
+    "SingleUseActivityManager",
 )
diff --git a/golem/managers/work/__init__.py b/golem/managers/work/__init__.py
@@ -1,9 +1,11 @@
 from golem.managers.work.asynchronous import AsynchronousWorkManager
+from golem.managers.work.mixins import WorkManagerPluginsMixin
 from golem.managers.work.plugins import redundancy_cancel_others_on_first_done, retry, work_plugin
 from golem.managers.work.sequential import SequentialWorkManager
 
 __all__ = (
     "AsynchronousWorkManager",
+    "WorkManagerPluginsMixin",
     "SequentialWorkManager",
     "work_plugin",
     "redundancy_cancel_others_on_first_done",

diff --git a/tests/unit/test_mid.py → tests/unit/test_pipeline.py b/tests/unit/test_mid.py → tests/unit/test_pipeline.py