Using typing-extensions for 3.9 support of @OverRide

Author: unkcpz

pyproject.toml

@@ -32,6 +32,7 @@ dependencies = [
     'kiwipy[rmq]~=0.8.5',
     'nest_asyncio~=1.5,>=1.5.1',
     'pyyaml~=6.0',
+    'typing-extensions~=4.12'
 ]
 
 [project.urls]

src/plumpy/base/state_machine.py

@@ -275,6 +275,8 @@ def state(self) -> State | None:
     def state_label(self) -> Any:
         if self._state is None:
             return None
+        # XXX: should not use `.value` to access the printable output from LABEL
+        # LABEL as the ClassVar should have __str__
         return self._state.LABEL
 
     def add_state_event_callback(self, hook: Hashable, callback: EVENT_CALLBACK_TYPE) -> None:

src/plumpy/persistence.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from __future__ import annotations
+
 import abc
 import asyncio
 import collections

src/plumpy/process_states.py

@@ -17,10 +17,10 @@
     Union,
     cast,
     final,
-    override,
 )
 
 import yaml
+from typing_extensions import override
 from yaml.loader import Loader
 
 from plumpy.message import MessageBuilder, MessageType

src/plumpy/processes.py

@@ -285,20 +285,20 @@ def recreate_from(
         if 'loop' in load_context:
             proc._loop = load_context.loop
         else:
-            _LOGGER.warning(f'cannot find `loop` store in load_context, use default event loop')
+            _LOGGER.warning('cannot find `loop` store in load_context, use default event loop')
             proc._loop = asyncio.get_event_loop()
 
         proc._state = proc.recreate_state(saved_state['_state'])
 
         if 'coordinator' in load_context:
             proc._coordinator = load_context.coordinator
         else:
-            _LOGGER.warning(f'cannot find `coordinator` store in load_context')
+            _LOGGER.warning('cannot find `coordinator` store in load_context')
 
         if 'logger' in load_context:
             proc._logger = load_context.logger
         else:
-            _LOGGER.warning(f'cannot find `logger` store in load_context')
+            _LOGGER.warning('cannot find `logger` store in load_context')
 
         # Need to call this here as things downstream may rely on us having the runtime variable above
         persistence.load_auto_persist_params(proc, saved_state, load_context)
@@ -760,7 +760,9 @@ def on_entered(self, from_state: Optional[state_machine.State]) -> None:
             try:
                 self._coordinator.broadcast_send(body=None, sender=self.pid, subject=subject)
             except exceptions.CoordinatorCommunicationError:
-                message = f'Process<{self.pid}>: cannot broadcast state change from {from_label} to {self.state.value}'
+                message = (
+                    f'Process<{self.pid}>: cannot broadcast state change from {from_label} to {self.state_label.value}'
+                )
                 self.logger.warning(message)
                 self.logger.debug(message, exc_info=True)
             except Exception: