Fix text classification for community (#288)

Author: osanseviero

api-inference-community/docker_images/common/app/pipelines/text_classification.py

@@ -22,7 +22,7 @@ def __call__(self, inputs: str) -> List[Dict[str, float]]:
             inputs (:obj:`str`):
                 a string containing some text
         Return:
-            A :obj:`list`:. The object returned should be like [{"label": 0.9939950108528137}] containing :
+            A :obj:`list`:. The object returned should be a list of one list like [[{"label": 0.9939950108528137}]] containing:
                 - "label": A string representing what the label/class is. There can be multiple labels.
                 - "score": A score between 0 and 1 describing how confident the model is for this label/class.
         """

api-inference-community/docker_images/common/tests/test_api_text_classification.py

@@ -49,8 +49,10 @@ def test_simple(self):
         )
         content = json.loads(response.content)
         self.assertEqual(type(content), list)
+        self.assertEqual(len(content), 1)
+        self.assertEqual(type(content[0]), list)
         self.assertEqual(
-            set(k for el in content for k in el.keys()),
+            set(k for el in content[0] for k in el.keys()),
             {"label", "score"},
         )
 
@@ -63,8 +65,10 @@ def test_simple(self):
         )
         content = json.loads(response.content)
         self.assertEqual(type(content), list)
+        self.assertEqual(len(content), 1)
+        self.assertEqual(type(content[0]), list)
         self.assertEqual(
-            set(k for el in content for k in el.keys()),
+            set(k for el in content[0] for k in el.keys()),
             {"label", "score"},
         )
 

api-inference-community/docker_images/spacy/app/pipelines/text_classification.py

@@ -28,13 +28,13 @@ def __init__(
 
         self.model = spacy.load(model_name)
 
-    def __call__(self, inputs: str) -> List[Dict[str, float]]:
+    def __call__(self, inputs: str) -> List[List[Dict[str, float]]]:
         """
         Args:
             inputs (:obj:`str`):
                 a string containing some text
         Return:
-            A :obj:`list`:. The object returned should be like [{"label": 0.9939950108528137}] containing :
+            A :obj:`list`:. The object returned should be a list of one list like [[{"label": 0.9939950108528137}]] containing :
                 - "label": A string representing what the label/class is. There can be multiple labels.
                 - "score": A score between 0 and 1 describing how confident the model is for this label/class.
         """
@@ -44,4 +44,4 @@ def __call__(self, inputs: str) -> List[Dict[str, float]]:
         for cat, score in doc.cats.items():
             categories.append({"label": cat, "score": score})
 
-        return categories
+        return [categories]

api-inference-community/docker_images/spacy/tests/test_api_text_classification.py

@@ -49,8 +49,10 @@ def test_simple(self):
         )
         content = json.loads(response.content)
         self.assertEqual(type(content), list)
+        self.assertEqual(len(content), 1)
+        self.assertEqual(type(content[0]), list)
         self.assertEqual(
-            set(k for el in content for k in el.keys()),
+            set(k for el in content[0] for k in el.keys()),
             {"label", "score"},
         )
 
@@ -63,8 +65,10 @@ def test_simple(self):
         )
         content = json.loads(response.content)
         self.assertEqual(type(content), list)
+        self.assertEqual(len(content), 1)
+        self.assertEqual(type(content[0]), list)
         self.assertEqual(
-            set(k for el in content for k in el.keys()),
+            set(k for el in content[0] for k in el.keys()),
             {"label", "score"},
         )