Fix numpy 1.24 compatibility

Signed-off-by: Keith Battocchi <[email protected]>

Author: kbattocchi

econml/_ensemble/_ensemble.py

@@ -158,8 +158,8 @@ def _partition_estimators(n_estimators, n_jobs):
 
     # Partition estimators between jobs
     n_estimators_per_job = np.full(n_jobs, n_estimators // n_jobs,
-                                   dtype=np.int)
-    n_estimators_per_job[:n_estimators % n_jobs] += 1
+                                   dtype=int)
+    n_estimators_per_job[: n_estimators % n_jobs] += 1
     starts = np.cumsum(n_estimators_per_job)
 
     return n_jobs, n_estimators_per_job.tolist(), [0] + starts.tolist()

setup.cfg

@@ -58,6 +58,8 @@ tf =
     tensorflow > 1.10, < 2.3;python_version < '3.9'
     ; Version capped due to tensorflow incompatibility
     protobuf < 4
+    ; Version capped due to tensorflow incompatibility
+    numpy < 1.24
 plt =
     graphviz
     ; Version capped due to shap incompatibility
@@ -70,6 +72,8 @@ all =
     tensorflow > 1.10, < 2.3
     ; Version capped due to tensorflow incompatibility
     protobuf < 4
+    ; Version capped due to tensorflow incompatibility
+    numpy < 1.24
     ; Version capped due to shap incompatibility
     matplotlib < 3.6.0
     dowhy < 0.9