Skip to content

Commit aaf912f

Browse files
authored
Fix deprecation warning force_all_finite -> ensure_all_finite
1 parent 9e14790 commit aaf912f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cebra/integrations/sklearn/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def check_input_array(X: npt.NDArray, *, min_samples: int) -> npt.NDArray:
8181
dtype=("float16", "float32", "float64"),
8282
order=None,
8383
copy=False,
84-
force_all_finite=True,
84+
ensure_all_finite=True,
8585
ensure_2d=True,
8686
allow_nd=False,
8787
ensure_min_samples=min_samples,
@@ -112,7 +112,7 @@ def check_label_array(y: npt.NDArray, *, min_samples: int):
112112
dtype="numeric",
113113
order=None,
114114
copy=False,
115-
force_all_finite=True,
115+
ensure_all_finite=True,
116116
ensure_2d=False,
117117
allow_nd=False,
118118
ensure_min_samples=min_samples,

0 commit comments

Comments
 (0)