updated viz and ingestion tests

luxonis · Jan 14, 2025 · 2df74e6 · 2df74e6
1 parent 7307152
commit 2df74e6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_data/test_task_ingestion.py b/tests/test_data/test_task_ingestion.py
@@ -193,6 +193,7 @@ def generator4():
             path = make_image(i)
             yield {
                 "file": str(path),
+                "task": "detection",
                 "annotation": {
                     "class": "bike",
                     "boundingbox": {"x": 0.9, "y": 0.8, "w": 0.1, "h": 0.4},

diff --git a/tests/test_data/test_utils/test_visualizations.py b/tests/test_data/test_utils/test_visualizations.py
@@ -210,8 +210,7 @@ def test_visualize():
     expected_semantic = np.stack([expected_semantic] * 3, axis=-1)
     expected_images = {
         "image": image.copy(),
-        "semantic": expected_semantic,
-        "labels": expected_labels,
+        "labels": expected_semantic + expected_labels,
     }
     class_names = {
         "task": ["class_name"],