tetst new hp types

zauberzeug · Jan 12, 2025 · 6423b3c · 6423b3c
1 parent c007777
commit 6423b3c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 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/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)