Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

list -> generator

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-28 00:05:37 +02:00
parent 48737d9439
commit e0ab365bb9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/unit_tests/extractor_classifier_test.py
View File

@ -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