Fix mlflow package 2.16.0 release caused NCD gates/tests failing in I…

…nt/Canary (#3377)
Azure · Sep 10, 2024 · 28351c9 · 28351c9
1 parent 5782330
commit 28351c9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/cli/endpoints/online/ncd/lightgbm-iris/model/conda.yaml b/cli/endpoints/online/ncd/lightgbm-iris/model/conda.yaml
@@ -9,4 +9,5 @@ dependencies:
   - pandas==2.0.0
   - psutil==5.9.4
   - typing-extensions==4.5.0
+  - numpy==1.26.4
 name: mlflow-env
diff --git a/cli/endpoints/online/ncd/sklearn-diabetes/model/conda.yaml b/cli/endpoints/online/ncd/sklearn-diabetes/model/conda.yaml
@@ -8,4 +8,5 @@ dependencies:
   - scikit-learn==1.2.2
   - cloudpickle==2.2.1
   - psutil==5.9.4
+  - numpy==1.26.4
 name: mlflow-env