prevent wandb version error

opendilab · Apr 2, 2024 · cdd5ff4 · cdd5ff4
1 parent d7ccc9f
commit cdd5ff4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/unit_test.yml b/.github/workflows/unit_test.yml
@@ -26,6 +26,11 @@ jobs:
           python -m pip install .
           python -m pip install ".[test,k8s]"
           python -m pip install transformers
+          if python --version | grep -q "Python 3.7"; then
+              pip install wandb==0.16.4
+          else
+              echo "Python version is not 3.7, skipping wandb installation"
+          fi
           ./ding/scripts/install-k8s-tools.sh
           make unittest
       - name: Upload coverage to Codecov