learning-loop-node 0.13.1__tar.gz → 0.13.4__tar.gz

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: learning-loop-node
-Version: 0.13.1
+Version: 0.13.4
 Summary: Python Library for Nodes which connect to the Zauberzeug Learning Loop
 Home-page: https://github.com/zauberzeug/learning_loop_node
 License: MIT

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/learning_loop_node/detector/detector_node.py

@@ -200,8 +200,11 @@ class DetectorNode(Node):
         """The DetectorNode acts as a SocketIO server. This method sets up the server and defines the event handlers."""
         # pylint: disable=unused-argument
 
-        # Initialize the Socket.IO server
-        self.sio = socketio.AsyncServer(async_mode='asgi')
+        # Initialize the Socket.IO server with 20MB buffer size
+        self.sio = socketio.AsyncServer(
+            async_mode='asgi',
+            max_http_buffer_size=2e7,  # 20MB
+        )
         # Initialize and mount the ASGI app
         self.sio_app = socketio.ASGIApp(self.sio, socketio_path='/socket.io')
         self.mount('/ws', self.sio_app)
@@ -217,7 +220,8 @@ class DetectorNode(Node):
                     camera_id=data.get('camera-id', None) or data.get('mac', None),
                     tags=data.get('tags', []),
                     source=data.get('source', None),
-                    autoupload=data.get('autoupload', None)
+                    autoupload=data.get('autoupload', None),
+                    creation_date=data.get('creation_date', None)
                 )
                 if det is None:
                     return {'error': 'no model loaded'}
@@ -400,6 +404,7 @@ class DetectorNode(Node):
 
         self.log_status_on_change(status.state or 'None', status)
 
+        # NOTE: sending organization and project is no longer required!
         response = await self.sio_client.call('update_detector', (self.organization, self.project, jsonable_encoder(asdict(status))))
         if not response:
             self.socket_connection_broken = True

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/learning_loop_node/detector/inbox_filter/cam_observation_history.py

@@ -1,8 +1,8 @@
 import os
 from typing import List, Union
 
-from learning_loop_node.data_classes import (BoxDetection, ClassificationDetection, ImageMetadata, Observation,
-                                             PointDetection, SegmentationDetection)
+from ...data_classes import (BoxDetection, ClassificationDetection, ImageMetadata, Observation, PointDetection,
+                             SegmentationDetection)
 
 
 class CamObservationHistory:

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/learning_loop_node/detector/outbox.py

@@ -79,10 +79,10 @@ class Outbox():
         image_metadata.source = source or 'unknown'
         os.makedirs(tmp, exist_ok=True)
 
-        with open(tmp + '/image.json', 'w') as f:
+        with open(tmp + f'/image_{identifier}.json', 'w') as f:
             json.dump(jsonable_encoder(asdict(image_metadata)), f)
 
-        with open(tmp + '/image.jpg', 'wb') as f:
+        with open(tmp + f'/image_{identifier}.jpg', 'wb') as f:
             f.write(image)
 
         if os.path.exists(tmp):
@@ -141,8 +141,10 @@ class Outbox():
         # results in a post failure on the first run of the test in a docker environment (WTF)
 
         data: List[Tuple[str, Union[TextIOWrapper, BufferedReader]]] = []
-        data = [('files', open(f'{item}/image.json', 'r')) for item in items]
-        data += [('files', open(f'{item}/image.jpg', 'rb')) for item in items]
+        for item in items:
+            identifier = os.path.basename(item)
+            data.append(('files', open(f'{item}/image_{identifier}.json', 'r')))
+            data.append(('files', open(f'{item}/image_{identifier}.jpg', 'rb')))
 
         try:
             async with aiohttp.ClientSession() as session:
@@ -158,7 +160,11 @@ class Outbox():
         if response.status == 200:
             self.upload_counter += len(items)
             for item in items:
-                shutil.rmtree(item, ignore_errors=True)
+                try:
+                    shutil.rmtree(item)
+                    self.log.debug('Deleted %s', item)
+                except Exception:
+                    self.log.exception('Failed to delete %s', item)
             self.log.info('Uploaded %s images successfully', len(items))
 
         elif response.status == 422:

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/learning_loop_node/tests/trainer/conftest.py

@@ -36,8 +36,8 @@ async def test_initialized_trainer_node():
                                                 'model_variant': '',
                                                 'hyperparameters': {
                                    'resolution': 800,
-                                   'flip_rl': False,
-                                   'flip_ud': False}
+                                   'fliplr': 0.5,
+                                   'flipud': 0.5}
     })
     await node._on_startup()
     yield node
@@ -59,8 +59,8 @@ async def test_initialized_trainer():
                                                 'model_variant': '',
                                                 'hyperparameters': {
                                    'resolution': 800,
-                                   'flip_rl': False,
-                                   'flip_ud': False}
+                                   'fliplr': 0.5,
+                                   'flipud': 0.5}
     })
     yield trainer
     try:

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/learning_loop_node/trainer/trainer_logic_generic.py

@@ -402,7 +402,7 @@ class TrainerLogicGeneric(ABC):
         """
 
         files = await self._get_latest_model_files()
-        if files is None:
+        if files is None or len(files) == 0:
             raise CriticalError('Could not get latest model files. Training might have failed.')
 
         if isinstance(files, List):

{learning_loop_node-0.13.1 → learning_loop_node-0.13.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "learning_loop_node"
-version = "v0.13.1"
+version = "v0.13.4"
 description = "Python Library for Nodes which connect to the Zauberzeug Learning Loop"
 authors = ["Zauberzeug GmbH <info@zauberzeug.com>"]
 license = "MIT"