From e0ab365bb9179aaedf0f995c66c16d569bc979a7 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Mon, 28 Mar 2022 00:05:37 +0200 Subject: [PATCH] list -> generator --- test/unit_tests/extractor_classifier_test.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/test/unit_tests/extractor_classifier_test.py b/test/unit_tests/extractor_classifier_test.py index 7134010..9ce37d0 100644 --- a/test/unit_tests/extractor_classifier_test.py +++ b/test/unit_tests/extractor_classifier_test.py @@ -10,7 +10,6 @@ from image_prediction.extractor_classifier.extractor_classifier import Extractor @pytest.mark.parametrize("batch_size", [0, 1, 2, 16, 32, 64]) def test_extractor_classifier(image_extractor, image_classifier, images, expected_predictions): extractor_classifier = ExtractorClassifier(image_extractor, image_classifier) - results = list(extractor_classifier(images)) - print(results) + results = extractor_classifier(images) labels = list(map(itemgetter("label"), results)) assert labels == expected_predictions