From c3c9b8440944c93d03305a3694b8004f58da1c8a Mon Sep 17 00:00:00 2001 From: Matthias Date: Tue, 15 Apr 2025 07:26:32 +0200 Subject: [PATCH] chore: fix compatibility with numpy2 --- datasieve/transforms/dissimilarity_index.py | 2 +- datasieve/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/datasieve/transforms/dissimilarity_index.py b/datasieve/transforms/dissimilarity_index.py index 077a7ea..390a7a0 100644 --- a/datasieve/transforms/dissimilarity_index.py +++ b/datasieve/transforms/dissimilarity_index.py @@ -35,7 +35,7 @@ def fit(self, X, y=None, sample_weight=None, feature_list=None, **kwargs): pairwise = pairwise_distances(X) # remove the diagonal distances which are itself distances ~0 - np.fill_diagonal(pairwise, np.NaN) + np.fill_diagonal(pairwise, np.nan) pairwise = pairwise.reshape(-1, 1) self.avg_mean_dist = pairwise[~np.isnan(pairwise)].mean() self.trained_data = X diff --git a/datasieve/utils.py b/datasieve/utils.py index 4530ba8..e4c10d9 100644 --- a/datasieve/utils.py +++ b/datasieve/utils.py @@ -56,7 +56,7 @@ def find_training_horizon(df: pd.DataFrame, target_horizon, test_pct=0.05, current_window_distances = pairwise_distances( current_window, metric="euclidean", n_jobs=8) # remove the diagonal distances which are itself distances ~0 - # np.fill_diagonal(current_window_distances, np.NaN) + # np.fill_diagonal(current_window_distances, np.nan) current_window_distances = current_window_distances.reshape(-1, 1) std_train_dist = current_window_distances[~np.isnan(current_window_distances)].std() distances_horizon_current_window = pairwise_distances(