fix(tests): collect all predictions in generator

uci-uav-forge · Jun 21, 2024 · 016d7ad · 016d7ad
1 parent dce51f3
commit 016d7ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/imaging/image_processor_tests.py b/tests/imaging/image_processor_tests.py
@@ -337,7 +337,7 @@ def test_irl_dataset(self, gen_confusion_matrices = True):
             prediction_list = []
 
         for img, ground_truth in zip(imgs, labels):
-            predictions = image_processor.process_image(img)
+            predictions = list(image_processor.process_image(img))
 
 
             if gen_confusion_matrices: