diff --git a/test/unit_tests/test_predictor.py b/test/unit_tests/test_predictor.py index f429475..522ec4b 100644 --- a/test/unit_tests/test_predictor.py +++ b/test/unit_tests/test_predictor.py @@ -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