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

refactor

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-03-21 13:36:23 +01:00
parent 130d0e8b23
commit 530de2ff89
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit_tests/test_predictor.py
View File

@ -5,7 +5,7 @@ def test_predict_pdf_works(predictor, test_pdf):
assert predictions["class"] == "formula"
probabilities = predictions["probabilities"]
# Floating point precision problem for output so test only that keys exist not the values
assert all (key in probabilities for key in ("formula", "other", "signature", "logo"))
assert all(key in probabilities for key in ("formula", "other", "signature", "logo"))
metadata = list(metadata)
metadata = dict(**metadata[0])
metadata.pop("document_filename") # temp filename cannot be tested