diff --git a/golem/managers/activity/__init__.py b/golem/managers/activity/__init__.py index 2a7a8976..39f7a284 100644 --- 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 index 3ebe7a4d..437c17a5 100644 --- 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 b/tests/unit/test_pipeline.py similarity index 100% rename from tests/unit/test_mid.py rename to tests/unit/test_pipeline.py