From 28351c9af7115151016a56759ffff9d40d254c84 Mon Sep 17 00:00:00 2001
From: ym11369 <65710487+ym11369@users.noreply.github.com>
Date: Wed, 11 Sep 2024 00:00:53 +0800
Subject: [PATCH] Fix mlflow package 2.16.0 release caused NCD gates/tests
 failing in Int/Canary (#3377)

---
 cli/endpoints/online/ncd/lightgbm-iris/model/conda.yaml    | 1 +
 cli/endpoints/online/ncd/sklearn-diabetes/model/conda.yaml | 1 +
 2 files changed, 2 insertions(+)

diff --git a/cli/endpoints/online/ncd/lightgbm-iris/model/conda.yaml b/cli/endpoints/online/ncd/lightgbm-iris/model/conda.yaml
index 373bbbc351d..1af51bb414f 100644
--- 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
index 8c0ee860974..90b39dc1530 100644
--- 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