From d251eaa6ac168231d3ebbbb8dd91767115cc5f2b Mon Sep 17 00:00:00 2001 From: eschmidt42 <11818904+eschmidt42@users.noreply.github.com> Date: Sat, 16 Aug 2025 15:31:57 +0200 Subject: [PATCH] chore: fixed type errors in xgboost.py --- src/random_tree_models/xgboost.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/random_tree_models/xgboost.py b/src/random_tree_models/xgboost.py index 11f68f5..a228546 100644 --- a/src/random_tree_models/xgboost.py +++ b/src/random_tree_models/xgboost.py @@ -106,7 +106,7 @@ def xgboost_histogrammify_with_h( rank_bin_edges = np.histogram_bin_edges(rank, bins=n_bins) bin_assignments = pd.cut( - rank, bins=rank_bin_edges, labels=False, include_lowest=True + rank, bins=rank_bin_edges.tolist(), labels=False, include_lowest=True ) x_bin_edges = np.interp(rank_bin_edges, rank, x_ordered) @@ -126,7 +126,7 @@ def xgboost_histogrammify_with_x_bin_edges( for i in range(X.shape[1]): bin_assignments = pd.cut( - X[:, i], bins=all_x_bin_edges[i], labels=False, include_lowest=True + X[:, i], bins=all_x_bin_edges[i].tolist(), labels=False, include_lowest=True ) X_hist[:, i] = bin_assignments