Add numpy 2.0 support

Author: Roja Reddy Sareddy

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,<=1.7.6
+xgboost>=1.6.2,<=2.1.0
 pillow>=10.0.1,<=11
 opentelemetry-proto==1.27.0
 protobuf==4.25.5

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

@@ -6,10 +6,10 @@ dependencies:
 - pip:
   - mlflow>=2.16.1
   - lz4==4.3.2
-  - numpy>=2.0.0
+  - numpy>=1.26.4
   - pandas==2.2.3
   - psutil==5.9.8
   - scikit-learn==1.6.1
   - scipy==1.13.0
-  - xgboost==1.7.1
+  - xgboost==2.1.0
 name: mlflow-env

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

@@ -1,8 +1,8 @@
 mlflow>=2.16.1
 lz4==4.3.2
-numpy>=2.0.0
+numpy>=1.26.4
 pandas==2.2.3
 psutil==5.9.8
 scikit-learn==1.6.1
 scipy==1.13.0
-xgboost==1.7.1
+xgboost==2.1.0