Fix spacy 2

Author: makseq

label_studio_ml/examples/nemo_asr/test_api.py

@@ -27,49 +27,7 @@ def client():
 def test_predict(client):
     request = json.loads('{"tasks": [{"id": 102651352, "data": {"audio": "https://htx-pub.s3.amazonaws.com/datasets/audio/f2bjrop1.1.wav"}, "meta": {}, "created_at": "2024-04-08T15:52:07.473205Z", "updated_at": "2024-04-08T15:52:07.473210Z", "is_labeled": false, "overlap": 1, "inner_id": 2, "total_annotations": 0, "cancelled_annotations": 0, "total_predictions": 0, "comment_count": 0, "unresolved_comment_count": 0, "last_comment_updated_at": null, "project": 62075, "updated_by": null, "file_upload": 1430391, "comment_authors": [], "annotations": [], "predictions": []}], "project": "62075.1712591487", "label_config": "<View>\\n  <Audio name=\\"audio\\" value=\\"$audio\\" zoom=\\"true\\" hotkey=\\"ctrl+enter\\" />\\n  <Header value=\\"Provide Transcription\\" />\\n  <TextArea name=\\"transcription\\" toName=\\"audio\\"\\n            rows=\\"4\\" editable=\\"true\\" maxSubmissions=\\"1\\" />\\n</View>", "params": {"login": null, "password": null, "context": null}}')
 
-    expected_response = {
-        "results": [
-            {
-            "model_version": "SpacyMLBackend-v0.0.1",
-            "result": [
-                {
-                "from_name": "ner_tags",
-                "to_name": "text",
-                "type": "labels",
-                "value": {
-                    "end": 10,
-                    "labels": ["PERSON"],
-                    "start": 0,
-                    "text": "Katy Perry"
-                }
-                },
-                {
-                "from_name": "ner_tags",
-                "to_name": "text",
-                "type": "labels",
-                "value": {
-                    "end": 23,
-                    "labels": ["ORG"],
-                    "start": 15,
-                    "text": "Pharrell"
-                }
-                },
-                {
-                "from_name": "ner_tags",
-                "to_name": "text",
-                "type": "labels",
-                "value": {
-                    "end": 83,
-                    "labels": ["PRODUCT"],
-                    "start": 47,
-                    "text": "Apple Watches http://t.co/k6k3SdwShP"
-                }
-                }
-            ],
-            "score": 0.0
-            }
-        ]
-    }
+    expected_response = json.loads('{"results":[{"model_version":"NemoASR-v0.0.1","result":[{"from_name":"transcription","to_name":"audio","type":"textarea","value":{"text":["ffected to be named in march it may be the most important appointment governor michael dukakis makes during the remainder of his administration and one of the toughest s w b ar as murgr menikovi"]}}],"score":1.0}]}\n')
 
     response = client.post('/predict', data=json.dumps(request), content_type='application/json')
     assert response.status_code == 200

label_studio_ml/examples/spacy/test_api.py

@@ -13,6 +13,50 @@ def client():
 
 
 def test_predict(client):
+    expected_response = {
+        "results": [
+            {
+            "model_version": "SpacyMLBackend-v0.0.1",
+            "result": [
+                {
+                "from_name": "ner_tags",
+                "to_name": "text",
+                "type": "labels",
+                "value": {
+                    "end": 10,
+                    "labels": ["PERSON"],
+                    "start": 0,
+                    "text": "Katy Perry"
+                }
+                },
+                {
+                "from_name": "ner_tags",
+                "to_name": "text",
+                "type": "labels",
+                "value": {
+                    "end": 23,
+                    "labels": ["ORG"],
+                    "start": 15,
+                    "text": "Pharrell"
+                }
+                },
+                {
+                "from_name": "ner_tags",
+                "to_name": "text",
+                "type": "labels",
+                "value": {
+                    "end": 83,
+                    "labels": ["PRODUCT"],
+                    "start": 47,
+                    "text": "Apple Watches http://t.co/k6k3SdwShP"
+                }
+                }
+            ],
+            "score": 0.0
+            }
+        ]
+    }
+
     response = client.post('/predict', json=json.loads('{"tasks": [{"id": 1428, "data": {"text": "Katy Perry and Pharrell have the same taste in Apple Watches http://t.co/k6k3SdwShP"}, "meta": {}, "created_at": "2024-03-20T20:50:22.955810Z", "updated_at": "2024-03-20T20:50:22.955819Z", "is_labeled": false, "overlap": 1, "inner_id": 18, "total_annotations": 0, "cancelled_annotations": 0, "total_predictions": 0, "comment_count": 0, "unresolved_comment_count": 0, "last_comment_updated_at": null, "project": 6, "updated_by": null, "file_upload": 8, "comment_authors": [], "annotations": [], "predictions": []}], "project": "6.1710967805", "label_config": "<View>\\n  <Labels name=\\"ner_tags\\" toName=\\"text\\">\\n    <Label value=\\"label_1\\" background=\\"#FFA39E\\"/>\\n    <Label value=\\"label_2\\" background=\\"#D4380D\\"/>\\n    <Label value=\\"label_3\\" background=\\"#FFC069\\"/>\\n  </Labels>\\n  <Text name=\\"text\\" value=\\"$text\\"/>\\n</View>", "params": {"login": null, "password": null, "context": null}}'))
     assert response.status_code == 200
-    assert response.json == json.loads('{"results":[{"model_version":"SpacyMLBackend-v0.0.1","result":[{"from_name":"ner_tags","to_name":"text","type":"labels","value":{"end":10,"labels":["PERSON"],"start":0,"text":"Katy Perry"}},{"from_name":"ner_tags","to_name":"text","type":"labels","value":{"end":23,"labels":["ORG"],"start":15,"text":"Pharrell"}},{"from_name":"ner_tags","to_name":"text","type":"labels","value":{"end":60,"labels":["ORG"],"start":47,"text":"Apple Watches"}}],"score":0.0}]}\n')
+    assert response.json == expected_response