From 44a1ee008a44c62c479a6eadc1415b4cfd164058 Mon Sep 17 00:00:00 2001 From: ym11369 Date: Mon, 9 Sep 2024 18:29:45 +0800 Subject: [PATCH] Fix mlflow package 2.16.0 release caused NCD gates/tests failing in Int/Canary --- 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