mis

Author: unkcpz

src/plumpy/__init__.py

@@ -26,9 +26,9 @@
     PersistenceError,
     UnsuccessfulResult,
 )
-from .futures import CancellableAction, Future, capture_exceptions, create_task
+from .futures import CancellableAction, Future, capture_exceptions
 from .loaders import DefaultObjectLoader, ObjectLoader, get_object_loader, set_object_loader
-from .message import MessageBuilder, ProcessLauncher, create_continue_body, create_launch_body
+from .message import MsgContinue, MsgCreate, MsgKill, MsgLaunch, MsgPause, MsgPlay, MsgStatus, ProcessLauncher
 from .persistence import (
     Bundle,
     InMemoryPersister,
@@ -64,26 +64,17 @@
 from .workchains import ToContext, WorkChain, WorkChainSpec, if_, return_, while_
 
 __all__ = (
-    # ports
     'UNSPECIFIED',
-    # utils
     'AttributesDict',
-    # persistence
     'Bundle',
-    # processes
     'BundleKeys',
-    # futures
     'CancellableAction',
-    # exceptions
     'ClosedError',
-    # process_states/States
     'Continue',
-    # coordinator
     'Coordinator',
     'CoordinatorConnectionError',
     'CoordinatorTimeoutError',
     'Created',
-    # loaders
     'DefaultObjectLoader',
     'Excepted',
     'Finished',
@@ -92,39 +83,39 @@
     'InputPort',
     'Interruption',
     'InvalidStateError',
-    # process_states/Commands
     'Kill',
     'KillInterruption',
     'Killed',
     'KilledError',
     'LoadSaveContext',
-    # message
-    'MessageBuilder',
+    'MsgContinue',
+    'MsgCreate',
+    'MsgKill',
+    'MsgLaunch',
+    'MsgPause',
+    'MsgPlay',
+    'MsgStatus',
     'ObjectLoader',
     'OutputPort',
     'PauseInterruption',
     'PersistedCheckpoint',
     'PersistenceError',
     'Persister',
     'PicklePersister',
-    # event
     'PlumpyEventLoopPolicy',
     'Port',
     'PortNamespace',
     'PortValidationError',
     'Process',
-    # controller
     'ProcessController',
     'ProcessLauncher',
-    # process_listener
     'ProcessListener',
     'ProcessSpec',
     'ProcessState',
     'Running',
     'Savable',
     'SavableFuture',
     'Stop',
-    # workchain
     'ToContext',
     'TransitionFailed',
     'UnsuccessfulResult',
@@ -136,7 +127,6 @@
     'capture_exceptions',
     'create_continue_body',
     'create_launch_body',
-    'create_task',
     'get_event_loop',
     'get_object_loader',
     'if_',