fixed test

luxonis · Jan 16, 2025 · 37fed81 · 37fed81
1 parent dd831b0
commit 37fed81
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_data/test_dataset.py b/tests/test_data/test_dataset.py
@@ -516,11 +516,12 @@ def generator1():
 
 
 def test_merge_datasets(
-    dataset_name: str,
     bucket_storage: BucketStorage,
+    dataset_name: str,
     tempdir: Path,
     subtests: SubTests,
 ):
+    dataset_name = f"{dataset_name}_{bucket_storage.value}"
     dataset1_name = f"{dataset_name}_1"
     dataset1 = LuxonisDataset(
         dataset1_name,
@@ -583,8 +584,8 @@ def generator2():
             new_dataset_name=f"{dataset1_name}_{dataset2_name}_merged",
         )
 
-        classes = dataset1_merged_with_dataset2.get_classes()
-        assert set(classes[""]) == {"person", "dog"}
+    classes = dataset1_merged_with_dataset2.get_classes()
+    assert set(classes[""]) == {"person", "dog"}
 
     df_merged = dataset1_merged_with_dataset2._load_df_offline()
     df_cloned_merged = dataset1.merge_with(