Merge pull request #117 from iryna-kondr/mlc_fix

fixed mlc labels collection

Author: OKUA1

pyproject.toml

@@ -11,7 +11,7 @@ dependencies = [
   "google-cloud-aiplatform[pipelines]>=1.27.0,<2.0.0"
 ]
 name = "scikit-llm"
-version = "1.4.0"
+version = "1.4.1"
 authors = [
   { name="Oleh Kostromin", email="[email protected]" },
   { name="Iryna Kondrashchenko", email="[email protected]" },

skllm/__init__.py

@@ -1,2 +1,2 @@
-__version__ = '1.4.0'
+__version__ = '1.4.1'
 __author__ = 'Iryna Kondrashchenko, Oleh Kostromin'

skllm/models/_base/classifier.py

@@ -137,8 +137,11 @@ def _extract_labels(self, y) -> List[str]:
         """
         labels = []
         for l in y:
-            for j in l:
-                labels.append(j)
+            if isinstance(l, list):
+                for j in l:
+                    labels.append(j)
+            else:
+                labels.append(l)
         return labels