run_tests_with_gpu.yml

name: Run Tests with GPU

on:
  workflow_dispatch: # Allows for manual triggering

concurrency:
  group: run-tests-gpu # Fixed group name to ensure only one instance runs
  cancel-in-progress: false

jobs:
  test:
    runs-on: ubuntu-latest

    steps:
    - name: Set SSH permissions
      run: |
        mkdir -p ~/.ssh
        chmod 700 ~/.ssh
        sudo chown $USER:$USER ~/.ssh

    - name: Checkout code
      uses: actions/checkout@v3

    - name: Set up Python 3.10
      uses: actions/setup-python@v5
      with:
        python-version: "3.10"

    - name: Install and configure dstack
      run: |
        pip install dstack
        dstack config --url https://sky.dstack.ai --project ${{ secrets.DSTACK_PROJECT }} --token ${{ secrets.DSTACK_TOKEN }}
        dstack init
    
    - name: Run tests with GPU
      run: |
        HF_TOKEN=${{ secrets.HF_TOKEN }} dstack apply -f tests.dstack.yml --force -y

    - name: Extract pytest logs
      if: ${{ always() }}
      run: |
        dstack logs aana-tests | sed -n '/============================= test session starts ==============================/,$p'