learning-loop-node 0.19.0__tar.gz → 0.20.0__tar.gz

{learning_loop_node-0.19.0 → learning_loop_node-0.20.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: learning_loop_node
-Version: 0.19.0
+Version: 0.20.0
 Summary: Python Library for Nodes which connect to the Zauberzeug Learning Loop
 Author-email: Zauberzeug GmbH <info@zauberzeug.com>
 License: MIT

{learning_loop_node-0.19.0 → learning_loop_node-0.20.0}/learning_loop_node/trainer/trainer_logic.py

@@ -147,20 +147,12 @@ class TrainerLogic(TrainerLogicGeneric):
     async def stop(self) -> None:
         """If executor is running, stop it. Else cancel training task."""
         print('===============> stop received in trainer_logic.', flush=True)
-
         if not self.training_active:
             return
         if self._executor and self._executor.is_running():
             await self.executor.stop_and_wait()
-        elif self.training_task:
-            logging.info('cancelling training task')
-            if self.training_task.cancel():
-                try:
-                    await self.training_task
-                except asyncio.CancelledError:
-                    pass
-                logging.info('cancelled training task')
-                self._may_restart()
+            return
+        await super().stop()
 
     # ---------------------------------------- ABSTRACT METHODS ----------------------------------------
 

{learning_loop_node-0.19.0 → learning_loop_node-0.20.0}/learning_loop_node/trainer/trainer_logic_generic.py

@@ -470,18 +470,23 @@ class TrainerLogicGeneric(ABC):
         await self.stop()
 
     async def stop(self):
-        """Stops the training process by canceling training task.
+        """Stop the training. Default is to abort.
+        """
+        await self.abort()
+
+    async def abort(self):
+        """Abort the training: cancel the task immediately and discards in-progress model.
         """
         if not self.training_active:
             return
         if self.training_task:
-            logger.info('cancelling training task')
+            logger.info('aborting training task')
             if self.training_task.cancel():
                 try:
                     await self.training_task
                 except asyncio.CancelledError:
                     pass
-                logger.info('cancelled training task')
+                logger.info('aborted training task')
                 self._may_restart()
 
     def _may_restart(self) -> None:

{learning_loop_node-0.19.0 → learning_loop_node-0.20.0}/learning_loop_node/trainer/trainer_node.py

@@ -76,6 +76,15 @@ class TrainerNode(Node):
                 self.log.exception('error in stop_training. Exception:')
             return True
 
+        @sio_client.event
+        async def abort_training():
+            self.log.info('abort_training received. Current state : %s', self.trainer_logic.state)
+            try:
+                await self.trainer_logic.abort()
+            except Exception:
+                self.log.exception('error in abort_training. Exception:')
+            return True
+
     async def send_status(self):
         if not self.sio_client or not self.sio_client.connected:
             self.log.debug('cannot send status - not connected to the Learning Loop')

{learning_loop_node-0.19.0 → learning_loop_node-0.20.0}/learning_loop_node.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: learning_loop_node
-Version: 0.19.0
+Version: 0.20.0
 Summary: Python Library for Nodes which connect to the Zauberzeug Learning Loop
 Author-email: Zauberzeug GmbH <info@zauberzeug.com>
 License: MIT

{learning_loop_node-0.19.0 → learning_loop_node-0.20.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "learning_loop_node"
-version = "0.19.0"
+version = "0.20.0"
 description = "Python Library for Nodes which connect to the Zauberzeug Learning Loop"
 authors = [{ name = "Zauberzeug GmbH", email = "info@zauberzeug.com" }]
 license = { text = "MIT" }