Merge branch 'main' into fix-batch-upload

zauberzeug · Jan 24, 2025 · 568f3d0 · 568f3d0
2 parents e5ed4f7 + 6c1f422
commit 568f3d0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/learning_loop_node/tests/trainer/conftest.py b/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:

diff --git a/learning_loop_node/trainer/trainer_logic_generic.py b/learning_loop_node/trainer/trainer_logic_generic.py
@@ -402,7 +402,7 @@ async def _upload_model_return_new_model_uuid(self, context: Context) -> str:
         """
 
         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):

diff --git a/mock_trainer/app_code/tests/test_detections.py b/mock_trainer/app_code/tests/test_detections.py
@@ -4,7 +4,6 @@
 # pylint: disable=protected-access,redefined-outer-name,unused-argument
 import pytest
 from fastapi.encoders import jsonable_encoder
-
 from learning_loop_node.data_classes import Category, Context
 from learning_loop_node.globals import GLOBALS
 from learning_loop_node.tests import test_helper
@@ -29,8 +28,8 @@ async def test_all():
                'model_variant': '',
                'hyperparameters': {
                    'resolution': 800,
-                   'flip_rl': False,
-                   'flip_ud': False}
+                   'fliplr': 0.5,
+                   'flipud': 0.5}
                }
     # await asyncio.sleep(100)