add type cast to cover iterables

basf · May 5, 2024 · eaeee46 · eaeee46
1 parent 7ed4b24
commit eaeee46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/molpipeline/estimators/chemprop/models.py b/molpipeline/estimators/chemprop/models.py
@@ -193,7 +193,7 @@ def fit(
             The fitted model.
         """
         if self._is_classifier():
-            self._classes_ = np.unique(y)
+            self._classes_ = np.unique(list(y))
         return super().fit(X, y)
 
     def predict(