Fix n_features_in_ in KNeighborsTransformer

GAA-UAM · Jun 28, 2023 · c7358a4 · c7358a4
1 parent 706db42
commit c7358a4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skfda/preprocessing/dim_reduction/_neighbor_transforms.py b/skfda/preprocessing/dim_reduction/_neighbor_transforms.py
@@ -102,7 +102,8 @@ def _fit(
         fit_with_zeros: bool = True,
     ) -> SelfType:
         ret = super()._fit(X, y)
-        self.n_features_in_ = self._estimator.n_features_in_
+
+        self.n_features_in_ = 1 if isinstance(X, FData) else X.shape[1]
 
         return ret