From c1b9117b69955bd8bb3d5fc2bf4cdb2374af0843 Mon Sep 17 00:00:00 2001 From: Joaquin Dominguez Date: Tue, 1 Apr 2025 11:36:28 -0400 Subject: [PATCH] Update hdbscan_.py --- hdbscan/hdbscan_.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hdbscan/hdbscan_.py b/hdbscan/hdbscan_.py index ee317d60..61d375fb 100644 --- a/hdbscan/hdbscan_.py +++ b/hdbscan/hdbscan_.py @@ -764,7 +764,7 @@ def hdbscan( # Checks input and converts to an nd-array where possible if metric != "precomputed" or issparse(X): - X = check_array(X, accept_sparse="csr", force_all_finite=False) + X = check_array(X, accept_sparse="csr", ensure_all_finite=False) else: # Only non-sparse, precomputed distance matrices are handled here # and thereby allowed to contain numpy.inf for missing distances @@ -1229,7 +1229,7 @@ def fit(self, X, y=None): if self.metric != "precomputed": # Non-precomputed matrices may contain non-finite values. # Rows with these values - X = check_array(X, accept_sparse="csr", force_all_finite=False) + X = check_array(X, accept_sparse="csr", ensure_all_finite=False) self._raw_data = X self._all_finite = is_finite(X)