fix tests

HumanSignal · Feb 21, 2025 · 2ac50db · 2ac50db
1 parent d6fe4d4
commit 2ac50db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/label_studio_ml/examples/gliner/test_api.py b/label_studio_ml/examples/gliner/test_api.py
@@ -56,11 +56,11 @@ def test_predict(client):
     }
 
     expected_response = {"results": [{"model_version": "GLiNERModel-v0.0.1", "result": [
-        {"from_name": "label", "score": 0.8706, "to_name": "text", "type": "labels",
+        {"from_name": "label", "score": 0.922, "to_name": "text", "type": "labels",
          "value": {"end": 11, "labels": ["Medication/Vaccine"], "start": 0, "text": "atomoxetine"}},
         {"from_name": "label", "score": 0.8706, "to_name": "text", "type": "labels",
          "value": {"end": 65, "labels": ["Medication/Vaccine"], "start": 32,
-                   "text": "norepinephrine reuptake inhibitor"}}], "score": 0.8706}]}
+                   "text": "norepinephrine reuptake inhibitor"}}], "score": 0.7053}]}
 
     response = client.post('/predict', data=json.dumps(request), content_type='application/json')
     assert response.status_code == 200