ci: Pin specific versions of numerical computation packages

VishwamAI · Oct 31, 2024 · 047d2e8 · 047d2e8
1 parent e7e677e
commit 047d2e8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/ci_cd.yml b/.github/workflows/ci_cd.yml
@@ -26,6 +26,8 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
+          # Install specific versions of numerical computation packages
+          pip install --no-deps numpy==1.23.5 scipy==1.9.0 jax==0.4.13 jaxlib==0.4.13 flax==0.6.10 optax==0.1.5
       - name: Create necessary directories
         run: |
           mkdir -p ${{ github.workspace }}/data