just tag to clone

.github/actions/setup-python-runner/action.yml

@@ -4,11 +4,24 @@ inputs:
   metabase-automation-token:
     description: Token for accessing the python-runner repository
     required: true
+  python-runner-version:
+    description: Git tag or commit of the python-runner repository to checkout
+    required: false
+    default: 'v0.0.1'
 runs:
   using: composite
   steps:
-    - name: Setup python-runner
-      run: ./.github/scripts/setup-python-runner.sh
+    - name: Clone and setup python-runner
+      run: |
+        echo "Cloning python-runner repository..."
+        git clone --depth 1 --branch ${{ inputs.python-runner-version }} \
+          https://${{ inputs.metabase-automation-token }}@github.com/metabase/python-runner-container.git python-runner
+
+        echo "Starting python-runner..."
+        cd python-runner
+        make run-ci 1> ../python-runner.out.log 2> ../python-runner.err.log
+        make logs 1>> ../python-runner.out.log 2>> ../python-runner.err.log &
+        cd ..
       shell: bash
       env:
         METABASE_AUTOMATION_USER_TOKEN: ${{ inputs.metabase-automation-token }}

.github/workflows/drivers.yml

@@ -6,6 +6,10 @@ on:
       skip:
         type: boolean
         default: false
+      python-runner-version:
+        type: string
+        default: 'v0.0.1'
+        description: 'Git tag or commit of python-runner repository to use'
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.head_ref && github.ref || github.run_id }}-drivers
@@ -138,6 +142,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Test BigQuery Cloud SDK driver
       id: test-driver
       uses: ./.github/actions/test-driver
@@ -193,6 +198,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Add ClickHouse TLS instance to /etc/hosts
       run: |
         sudo echo "127.0.0.1 server.clickhouseconnect.test" | sudo tee -a /etc/hosts
@@ -394,6 +400,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Test H2 driver
       id: test-driver
       uses: ./.github/actions/test-driver
@@ -516,6 +523,7 @@ jobs:
         uses: ./.github/actions/setup-python-runner
         with:
           metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+          python-runner-version: ${{ inputs.python-runner-version }}
       - name: Test ${{ matrix.version.name }}
         id: test-driver
         uses: ./.github/actions/test-driver
@@ -587,6 +595,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Test ${{ matrix.version.name }}
       id: test-driver
       uses: ./.github/actions/test-driver
@@ -856,6 +865,7 @@ jobs:
         uses: ./.github/actions/setup-python-runner
         with:
           metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+          python-runner-version: ${{ inputs.python-runner-version }}
       - name: Test ${{ matrix.version.name }} (${{ matrix.job.name }})
         id: test-driver
         uses: ./.github/actions/test-driver
@@ -994,6 +1004,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Test ${{ matrix.job.name }}
       id: test-driver
       if: ${{ !matrix.job.skip }}
@@ -1063,6 +1074,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Test Snowflake driver
       id: test-driver
       uses: ./.github/actions/test-driver
@@ -1199,6 +1211,7 @@ jobs:
       uses: ./.github/actions/setup-python-runner
       with:
         metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+        python-runner-version: ${{ inputs.python-runner-version }}
     - name: Test ${{ matrix.version.name }}
       id: test-driver
       uses: ./.github/actions/test-driver

.github/workflows/e2e-test.yml

@@ -28,6 +28,11 @@ on:
       split_index: # index of the chunk
         required: false
         type: number
+      python-runner-version:
+        type: string
+        default: 'v0.0.1'
+        description: 'Git tag or commit of python-runner repository to use'
+        required: false
 
 jobs:
   e2e-tests:
@@ -65,9 +70,10 @@ jobs:
           token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
 
       - name: Setup python-runner
-        env:
-          METABASE_AUTOMATION_USER_TOKEN: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
-        run: ./.github/scripts/setup-python-runner.sh
+        uses: ./.github/actions/setup-python-runner
+        with:
+          metabase-automation-token: ${{ secrets.METABASE_AUTOMATION_USER_TOKEN }}
+          python-runner-version: ${{ inputs.python-runner-version }}
 
       - name: Prepare Docker containers
         uses: ./.github/actions/e2e-prepare-containers
@@ -246,3 +252,9 @@ jobs:
              const summary = formatSummary(results);
 
              await core.summary.addRaw(summary).write();
+
+      - name: Cleanup python-runner
+        if: always()
+        uses: ./.github/actions/cleanup-python-runner
+        with:
+          upload-logs: false