adding python verion

databricks · Dec 24, 2024 · 010fc1d · 010fc1d
1 parent ceeb5c6
commit 010fc1d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -29,13 +29,15 @@ jobs:
           gpu_num: 1
           markers: "gpu"
           pip_deps: "[all]"
+          python_version: 3.11
           pytest_command: "coverage run -m pytest"
           ci_repo_gpu_test_ref: v0.2.2
         - name: "gpu2-llmfoundry2.5.1-1"
           container: mosaicml/llm-foundry:2.5.1_cu124-latest
           gpu_num: 2
           markers: "gpu"
           pip_deps: "[all]"
+          python_version: 3.11
           pytest_command: "coverage run -m pytest"
           ci_repo_gpu_test_ref: v0.2.2
     steps: