fix: allow sparse in DecayEstimator (#729)

koaning · Jan 13, 2025 · 9b15cef · 9b15cef
1 parent fef86eb
commit 9b15cef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklego/meta/decay_estimator.py b/sklego/meta/decay_estimator.py
@@ -124,7 +124,7 @@ def fit(self, X, y):
         """
 
         if self.check_input:
-            X, y = validate_data(self, X=X, y=y, dtype=FLOAT_DTYPES, reset=True)
+            X, y = validate_data(self, X=X, y=y, dtype=FLOAT_DTYPES, accept_sparse=True, reset=True)
         else:
             _check_n_features(self, X, reset=True)