From 005c6881ebaa64f7996639ed569da4de828c90ba Mon Sep 17 00:00:00 2001
From: Aditi Singh <47007177+Addi-11@users.noreply.github.com>
Date: Tue, 24 Dec 2024 22:10:04 +0530
Subject: [PATCH] vul fix (#3705)

* vul fix

* fix

* fix2
---
 .../environments/model-management/context/Dockerfile            | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/assets/training/model_management/environments/model-management/context/Dockerfile b/assets/training/model_management/environments/model-management/context/Dockerfile
index 8a48c22f6d..59108e025c 100644
--- a/assets/training/model_management/environments/model-management/context/Dockerfile
+++ b/assets/training/model_management/environments/model-management/context/Dockerfile
@@ -25,7 +25,7 @@ RUN conda install python=3.9 -y
 RUN pip install -r requirements.txt --no-cache-dir
 
 # Vulnerability fix
-RUN pip install Pillow gunicorn onnx==1.17.0 idna tqdm requests==2.32.1 tornado==6.4.2 certifi==2024.07.04 urllib3==1.26.19 scikit-learn==1.5.1
+RUN pip install Pillow gunicorn onnx==1.17.0 idna tqdm requests==2.32.1 tornado==6.4.2 certifi==2024.07.04 urllib3==1.26.19 scikit-learn==1.5.1 mlflow==2.19 mlflow-skinny==2.19
 
 # List installed packages
 RUN pip list