Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions cebra/integrations/sklearn/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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,
Expand Down
Loading