Add ability to specify python version

Eventual-Inc · Nov 20, 2024 · 32caf7d · 32caf7d
1 parent 2781ad9
commit 32caf7d
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/.github/assets/ray.yaml b/.github/assets/ray.yaml
@@ -48,8 +48,8 @@ setup_commands:
 - curl -LsSf https://astral.sh/uv/install.sh | sh
 - echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.bashrc
 - source ~/.bashrc
-- uv python install 3.9
-- uv python pin 3.9
+- uv python install <<PYTHON_VERSION>>
+- uv python pin <<PYTHON_VERSION>>
 - uv v
 - echo "source $HOME/.venv/bin/activate" >> $HOME/.bashrc
 - source .venv/bin/activate

diff --git a/.github/workflows/build-commit-run-tpch.yaml b/.github/workflows/build-commit-run-tpch.yaml
@@ -19,6 +19,10 @@ on:
         description: Which partition size to use
         required: false
         default: 2
+      python_version:
+        description: The version of python to use
+        required: false
+        default: '3.9'
 
 jobs:
   build:
@@ -34,3 +38,4 @@ jobs:
       skip_questions: ${{ inputs.skip_questions }}
       scale_factor: ${{ inputs.scale_factor }}
       partition_size: ${{ inputs.partition_size }}
+      python_version: ${{ inputs.python_version }}
diff --git a/.github/workflows/run-tpch.yaml b/.github/workflows/run-tpch.yaml
@@ -19,6 +19,10 @@ on:
         description: Which partition size to use
         required: false
         default: 2
+      python_version:
+        description: The version of python to use
+        required: false
+        default: '3.9'
   workflow_call:
     inputs:
       wheel:
@@ -41,11 +45,16 @@ on:
         description: Which partition size to use
         required: false
         default: 2
+      python_version:
+        type: string
+        description: The version of python to use
+        required: false
+        default: '3.9'
 
 jobs:
   run-tpch:
     runs-on: [self-hosted, linux, x64, ci-dev]
-    # timeout-minutes: 15 # Remove for ssh debugging
+    timeout-minutes: 15 # Remove for ssh debugging
     permissions:
       id-token: write
       contents: read
@@ -66,6 +75,7 @@ jobs:
         sed -i 's|<<WHEEL>>|${{ inputs.wheel }}|g' .github/assets/ray.yaml
         sed -i "s|<<SCALE_FACTOR>>|$scale_factor_str|g" .github/assets/ray.yaml
         sed -i 's|<<PARTITION_SIZE>>|${{ inputs.partition_size }}|g' .github/assets/ray.yaml
+        sed -i 's|<<PYTHON_VERSION>>|${{ inputs.python_version }}|g' .github/assets/ray.yaml
 
         # Download private ssh key
         KEY=$(aws secretsmanager get-secret-value --secret-id ci-github-actions-ray-cluster-key-3 --query SecretString --output text)