updated record tests

luxonis · Jan 14, 2025 · 9d8ade2 · 9d8ade2
1 parent 590fda8
commit 9d8ade2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_data/test_annotations.py b/tests/test_data/test_annotations.py
@@ -68,7 +68,7 @@ def compare_parquet_rows(
             {
                 "file": Path("tests/data/tempdir/left.jpg"),
                 "source_name": "image",
-                "task_name": "detection",
+                "task_name": "",
                 "class_name": None,
                 "instance_id": None,
                 "task_type": None,
@@ -90,7 +90,7 @@ def compare_parquet_rows(
             {
                 "file": Path("tests/data/tempdir/left.jpg"),
                 "source_name": "image",
-                "task_name": "detection",
+                "task_name": "",
                 "class_name": "person",
                 "instance_id": -1,
                 "task_type": "boundingbox",
@@ -99,7 +99,7 @@ def compare_parquet_rows(
             {
                 "file": Path("tests/data/tempdir/left.jpg"),
                 "source_name": "image",
-                "task_name": "detection",
+                "task_name": "",
                 "class_name": "person",
                 "instance_id": -1,
                 "task_type": "classification",