We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 437b6e9 commit 99a3ebdCopy full SHA for 99a3ebd
econml/_ensemble/_ensemble.py
@@ -158,8 +158,8 @@ def _partition_estimators(n_estimators, n_jobs):
158
159
# Partition estimators between jobs
160
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
+ dtype=int)
+ n_estimators_per_job[: n_estimators % n_jobs] += 1
163
starts = np.cumsum(n_estimators_per_job)
164
165
return n_jobs, n_estimators_per_job.tolist(), [0] + starts.tolist()
0 commit comments