Fix deprecation warning force_all_finite -> ensure_all_finite

AdaptiveMotorControlLab · Dec 17, 2024 · aaf912f · aaf912f
1 parent 9e14790
commit aaf912f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cebra/integrations/sklearn/utils.py b/cebra/integrations/sklearn/utils.py
@@ -81,7 +81,7 @@ def check_input_array(X: npt.NDArray, *, min_samples: int) -> npt.NDArray:
         dtype=("float16", "float32", "float64"),
         order=None,
         copy=False,
-        force_all_finite=True,
+        ensure_all_finite=True,
         ensure_2d=True,
         allow_nd=False,
         ensure_min_samples=min_samples,
@@ -112,7 +112,7 @@ def check_label_array(y: npt.NDArray, *, min_samples: int):
         dtype="numeric",
         order=None,
         copy=False,
-        force_all_finite=True,
+        ensure_all_finite=True,
         ensure_2d=False,
         allow_nd=False,
         ensure_min_samples=min_samples,