Updated type check logic

MinishLab · Feb 16, 2025 · 2dc5b17 · 2dc5b17
1 parent f9037d9
commit 2dc5b17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model2vec/train/classifier.py b/model2vec/train/classifier.py
@@ -232,7 +232,7 @@ def _initialize(self, y: LabelType) -> None:
         """
         if isinstance(y[0], (str, int)):
             # Check if all labels are strings or integers.
-            if not all(isinstance(label, (str | int)) for label in y):
+            if not all(isinstance(label, (str, int)) for label in y):
                 raise ValueError("Inconsistent label types in y. All labels must be strings or integers.")
             self.multilabel = False
             classes = sorted(set(y))