diff --git a/.circleci/config.yml b/.circleci/config.yml
index 7961cfddb..d95a8c214 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -690,6 +690,7 @@ jobs:
             pip install "respx==0.21.1"
             pip install "google-generativeai==0.3.2"
             pip install "google-cloud-aiplatform==1.43.0"
+            pip install "mlflow==2.17.2"
       # Run pytest and generate JUnit XML report
       - run:
           name: Run tests