casting categorical to array

intro-stat-learning · Mar 19, 2024 · 6beb419 · 6beb419
1 parent 6306805
commit 6beb419
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ISLP/models/model_spec.py b/ISLP/models/model_spec.py
@@ -259,8 +259,8 @@ def fit(self, X, y=None):
 
         self.column_info_ = _get_column_info(X,
                                              self.columns_,
-                                             self.is_categorical_,
-                                             self.is_ordinal_,
+                                             np.asarray(self.is_categorical_),
+                                             np.asarray(self.is_ordinal_),
                                              default_encoders=self.default_encoders)
         # include each column as a Feature
         # so that their columns are built if needed