Skip to content

Commit 2ac50db

Browse files
committed
fix tests
1 parent d6fe4d4 commit 2ac50db

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

label_studio_ml/examples/gliner/test_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,11 @@ def test_predict(client):
5656
}
5757

5858
expected_response = {"results": [{"model_version": "GLiNERModel-v0.0.1", "result": [
59-
{"from_name": "label", "score": 0.8706, "to_name": "text", "type": "labels",
59+
{"from_name": "label", "score": 0.922, "to_name": "text", "type": "labels",
6060
"value": {"end": 11, "labels": ["Medication/Vaccine"], "start": 0, "text": "atomoxetine"}},
6161
{"from_name": "label", "score": 0.8706, "to_name": "text", "type": "labels",
6262
"value": {"end": 65, "labels": ["Medication/Vaccine"], "start": 32,
63-
"text": "norepinephrine reuptake inhibitor"}}], "score": 0.8706}]}
63+
"text": "norepinephrine reuptake inhibitor"}}], "score": 0.7053}]}
6464

6565
response = client.post('/predict', data=json.dumps(request), content_type='application/json')
6666
assert response.status_code == 200

0 commit comments

Comments
 (0)