fix bug in using positional arg for reset

msamsami · Jan 16, 2025 · ddd3f09 · ddd3f09
1 parent 406487f
commit ddd3f09
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/wnb/_utils.py b/wnb/_utils.py
@@ -26,7 +26,7 @@ def validate_data(estimator, X, **kwargs: Any):
         return check_array(X, estimator=estimator, **kwargs)
 
     def _check_n_features(estimator, X, reset):
-        return estimator._check_n_features(X, reset)
+        return estimator._check_n_features(X, reset=reset)
 
-    def _check_feature_names(estimator, X, reset):
-        return estimator._check_feature_names(X, reset)
+    def _check_feature_names(estimator, X, *, reset):
+        return estimator._check_feature_names(X, reset=reset)