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_',

src/plumpy/process_states.py

@@ -23,8 +23,8 @@
 from typing_extensions import override
 from yaml.loader import Loader
 
-from plumpy.persistence import ensure_object_loader
 from plumpy.message import Message, MsgKill, MsgPause
+from plumpy.persistence import ensure_object_loader
 
 try:
     import tblib
@@ -638,7 +638,6 @@ class Killed:
     is_terminal: ClassVar[bool] = True
 
     def __init__(self, msg: Optional[Message]):
-
         """
         :param msg: Optional kill message
         """

src/plumpy/processes.py

@@ -1149,13 +1149,12 @@ def pause(self, msg_text: str | None = None) -> Union[bool, CancellableAction]:
 
         msg = MsgPause.new(msg_text)
         return self._do_pause(state_msg=msg)
-   
+
     @staticmethod
     def _interrupt(state: Interruptable, reason: Exception) -> None:
         state.interrupt(reason)
 
     def _do_pause(self, state_msg: Optional[Message], next_state: Optional[state_machine.State] = None) -> bool:
-
         """Carry out the pause procedure, optionally transitioning to the next state first"""
         try:
             if next_state is not None: