fix classification int test (#706)

predibase · Dec 6, 2024 · 61b40a8 · 61b40a8
1 parent 79e4283
commit 61b40a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integration-tests/test_classifications.py b/integration-tests/test_classifications.py
@@ -20,4 +20,4 @@ def test_distilbert_ner():
         )
         response.raise_for_status()
         print("RESPONSE FROM CLASSIFICATION: ", response.json())
-        assert len(response.json()["predictions"]) > 0
+        assert len(response.json()) > 0