diff --git a/src/tabpfn_extensions/interpretability/feature_selection.py b/src/tabpfn_extensions/interpretability/feature_selection.py index 5aff6cff..6dac0649 100644 --- a/src/tabpfn_extensions/interpretability/feature_selection.py +++ b/src/tabpfn_extensions/interpretability/feature_selection.py @@ -97,6 +97,7 @@ def _feature_selection( estimator, n_features_to_select=n_features_to_select, direction="forward", + **kwargs, ) sfs.fit(X, y) sfs.get_support()