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