learning-loop-node 0.18.4__tar.gz → 0.18.5__tar.gz

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: learning_loop_node
-Version: 0.18.4
+Version: 0.18.5
 Summary: Python Library for Nodes which connect to the Zauberzeug Learning Loop
 Author-email: Zauberzeug GmbH <info@zauberzeug.com>
 License: MIT
 Project-URL: Repository, https://github.com/zauberzeug/learning_loop_node
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: fastapi<1.0,>=0.128
 Requires-Dist: uvicorn[standard]<0.23.0,>=0.22.0
@@ -17,7 +17,7 @@ Requires-Dist: icecream<3.0.0,>=2.1.0
 Requires-Dist: aiofiles<0.8.0,>=0.7.0
 Requires-Dist: async_generator<2.0.0,>=1.10
 Requires-Dist: pytest-watch<5.0.0,>=4.2.0
-Requires-Dist: python-multipart<0.0.19,>=0.0.18
+Requires-Dist: python-multipart>=0.0.22
 Requires-Dist: psutil<6.0.0,>=5.8.0
 Requires-Dist: numpy<3.0,>=2.0
 Requires-Dist: Pillow<12.0.0,>=11.0.0

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/learning_loop_node/detector/detector_logic.py

@@ -38,10 +38,11 @@ class DetectorLogic():
             self.init()
             logging.info('Successfully loaded model %s', self.model_info)
             self._remaining_init_attempts = 2
-        except Exception:
+        except Exception as e:
             self._remaining_init_attempts -= 1
+            logging.error('Could not init model %s. Retries left: %s. Error: %s',
+                          self.model_info, self._remaining_init_attempts, e)
             self.model_info = None
-            logging.error('Could not init model %s. Retries left: %s', self.model_info, self._remaining_init_attempts)
             if self._remaining_init_attempts == 0:
                 raise NodeNeedsRestartError('Could not init model') from None
             raise

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/learning_loop_node/helpers/misc.py

@@ -60,11 +60,22 @@ def _handle_task_result(task: asyncio.Task,
 
 
 def get_free_memory_mb() -> float:  # NOTE used by yolov5
-    pynvml.nvmlInit()
-    h = pynvml.nvmlDeviceGetHandleByIndex(0)
-    info = pynvml.nvmlDeviceGetMemoryInfo(h)
-    free = float(info.free) / 1024 / 1024
-    return free
+    """
+    Get the free memory available to the graphics card (index 0) in MB.
+    If VRAM information is unavailable, for example if the system shares ram
+    between CPU and GPU, the available RAM is returned instead.
+    """
+    try:
+        pynvml.nvmlInit()
+
+        h = pynvml.nvmlDeviceGetHandleByIndex(0)
+        info = pynvml.nvmlDeviceGetMemoryInfo(h)
+        free_mb = float(info.free) / 1024 / 1024
+    except pynvml.nvml.NVMLError:
+        logger = logging.getLogger(__name__)
+        logger.warning('Could not get GPU memory info, returning free CPU memory instead')
+        free_mb = float(os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_AVPHYS_PAGES')) / 1024 / 1024
+    return free_mb
 
 
 async def is_valid_image(filename: str, check_jpeg: bool) -> bool:

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/learning_loop_node/trainer/io_helpers.py

@@ -173,7 +173,7 @@ class ActiveTrainingIO:
         response = await self.loop_communicator.post(
             f'/{context.organization}/projects/{context.project}/detections', json=detections_json)
         if response.status_code != 200:
-            msg = f'could not upload detections. {str(response)}'
+            msg = f'could not upload detections. HTTP {response.status_code}: {response.text}'
             logging.error(msg)
             raise Exception(msg)
 

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/learning_loop_node.egg-info/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: learning_loop_node
-Version: 0.18.4
+Version: 0.18.5
 Summary: Python Library for Nodes which connect to the Zauberzeug Learning Loop
 Author-email: Zauberzeug GmbH <info@zauberzeug.com>
 License: MIT
 Project-URL: Repository, https://github.com/zauberzeug/learning_loop_node
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: fastapi<1.0,>=0.128
 Requires-Dist: uvicorn[standard]<0.23.0,>=0.22.0
@@ -17,7 +17,7 @@ Requires-Dist: icecream<3.0.0,>=2.1.0
 Requires-Dist: aiofiles<0.8.0,>=0.7.0
 Requires-Dist: async_generator<2.0.0,>=1.10
 Requires-Dist: pytest-watch<5.0.0,>=4.2.0
-Requires-Dist: python-multipart<0.0.19,>=0.0.18
+Requires-Dist: python-multipart>=0.0.22
 Requires-Dist: psutil<6.0.0,>=5.8.0
 Requires-Dist: numpy<3.0,>=2.0
 Requires-Dist: Pillow<12.0.0,>=11.0.0

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/learning_loop_node.egg-info/requires.txt

@@ -8,7 +8,7 @@ icecream<3.0.0,>=2.1.0
 aiofiles<0.8.0,>=0.7.0
 async_generator<2.0.0,>=1.10
 pytest-watch<5.0.0,>=4.2.0
-python-multipart<0.0.19,>=0.0.18
+python-multipart>=0.0.22
 psutil<6.0.0,>=5.8.0
 numpy<3.0,>=2.0
 Pillow<12.0.0,>=11.0.0

{learning_loop_node-0.18.4 → learning_loop_node-0.18.5}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "learning_loop_node"
-version = "0.18.4"
+version = "0.18.5"
 description = "Python Library for Nodes which connect to the Zauberzeug Learning Loop"
 authors = [{ name = "Zauberzeug GmbH", email = "info@zauberzeug.com" }]
 license = { text = "MIT" }
 readme = "README.md"
-requires-python = ">=3.9"
+requires-python = ">=3.10"
 dependencies = [
     "fastapi>=0.128,<1.0",
     "uvicorn[standard]>=0.22.0,<0.23.0",
@@ -17,7 +17,7 @@ dependencies = [
     "aiofiles>=0.7.0,<0.8.0",
     "async_generator>=1.10,<2.0.0",
     "pytest-watch>=4.2.0,<5.0.0",
-    "python-multipart>=0.0.18,<0.0.19",
+    "python-multipart>=0.0.22",
     "psutil>=5.8.0,<6.0.0",
     "numpy>=2.0,<3.0",
     "Pillow>=11.0.0,<12.0.0",