Skip to content

Commit 99a3ebd

Browse files
committed
Fix numpy 1.24 compatibility
Signed-off-by: Keith Battocchi <kebatt@microsoft.com>
1 parent 437b6e9 commit 99a3ebd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

econml/_ensemble/_ensemble.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,8 +158,8 @@ def _partition_estimators(n_estimators, n_jobs):
158158

159159
# Partition estimators between jobs
160160
n_estimators_per_job = np.full(n_jobs, n_estimators // n_jobs,
161-
dtype=np.int)
162-
n_estimators_per_job[:n_estimators % n_jobs] += 1
161+
dtype=int)
162+
n_estimators_per_job[: n_estimators % n_jobs] += 1
163163
starts = np.cumsum(n_estimators_per_job)
164164

165165
return n_jobs, n_estimators_per_job.tolist(), [0] + starts.tolist()

0 commit comments

Comments
 (0)