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