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

refactoring; renaming

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-30 16:35:26 +02:00
parent a5d3232dd0
commit 0a22a35912
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit_tests/classifier_test.py
View File

@ -10,8 +10,8 @@ logger.setLevel(logging.DEBUG)
@pytest.mark.parametrize("estimator_type", ["mock", "keras"]) @pytest.mark.parametrize("estimator_type", ["mock", "keras"])
@pytest.mark.parametrize("label_format", ["index", "probability"]) @pytest.mark.parametrize("label_format", ["index", "probability"])
def test_predict(classifier, input_batch, expected_predictions_mapped): def test_classifier(classifier, input_batch, expected_predictions_mapped):
predictions = classifier.predict(input_batch) predictions = classifier(input_batch)
assert predictions == expected_predictions_mapped assert predictions == expected_predictions_mapped