diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 0f484a8..ef01b2a 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -26,13 +26,18 @@ jobs:
           - x64
     steps:
       - uses: actions/checkout@v2
+      - name: Setup custom python requirements
+        if: hashFiles('**/requirements.txt', '**/pyproject.toml') == ''
+        run: |
+          touch ./requirements.txt
+          echo "mlflow==2.13.0" > ./requirements.txt
       - uses: actions/setup-python@v4
         with:
           python-version: '3.10.13'
+          cache: 'pip'
       - name: Setup mlflow locally
         run: |
-          pip install mlflow
-          pip show mlflow 
+          pip install -r ./requirements.txt
           python3 /opt/hostedtoolcache/Python/3.10.13/x64/bin/mlflow server --host 0.0.0.0 --port 5000 &
           sleep 5
       - uses: julia-actions/setup-julia@v1