Add numpy 2.0 support

Author: Roja Reddy Sareddy

pyproject.toml

@@ -39,7 +39,7 @@ dependencies = [
   "google-pasta",
   "importlib-metadata>=1.4.0,<7.0",
   "jsonschema",
-  "numpy>=2.0.0,<2.3.0",
+  "numpy>=1.26.4,<2.3.0",
   "omegaconf>=2.2,<3",
   "packaging>=23.0,<25",
   "pandas",

requirements/extras/test_requirements.txt

@@ -29,7 +29,7 @@ cloudpickle==2.2.1
 jsonpickle<4.0.0
 PyYAML>=6.0.1
 # TODO find workaround
-xgboost>=1.6.2,<=2.1.0
+xgboost>=1.6.2,<=1.7.6
 pillow>=10.0.1,<=11
 opentelemetry-proto==1.27.0
 protobuf==4.25.5

tests/data/serve_resources/mlflow/xgboost/conda.yaml

@@ -11,5 +11,5 @@ dependencies:
   - psutil==5.9.8
   - scikit-learn==1.6.1
   - scipy==1.13.0
-  - xgboost==2.1.0
+  - xgboost==1.7.1
 name: mlflow-env

tests/data/serve_resources/mlflow/xgboost/requirements.txt

@@ -5,4 +5,4 @@ pandas==2.2.3
 psutil==5.9.8
 scikit-learn==1.6.1
 scipy==1.13.0
-xgboost==2.1.0
+xgboost==1.7.1