ci-pdm-install-and-test-gpu.yml

# cicd workflow for running tests with pytest
# needs to first install pdm, then install torch cpu manually and then install the package
# then run the tests

name: test (pdm install, gpu)

on: [push, pull_request]

jobs:
  tests:
    runs-on: "cirun-aws-runner--${{ github.run_id }}"
    steps:
      - name: Checkout
        uses: actions/checkout@v2

      - name: Set up Python 3.9
        uses: actions/setup-python@v2
        with:
          python-version: 3.9

      - name: Install pdm
        run: |
          python -m pip install pdm

      - name: Create venv
        run: |
          pdm config venv.in_project False
          pdm config venv.location /opt/dlami/nvme/venv
          pdm venv create --with-pip

      - name: Install torch (GPU CUDA 12.1)
        run: |
          pdm run python -m pip install torch  --index-url https://download.pytorch.org/whl/cu121

      - name: Print and check torch version
        run: |
          pdm run python -c "import torch; print(torch.__version__)"
          pdm run python -c "import torch; assert not torch.__version__.endswith('+cpu')"

      - name: Install package (including dev dependencies)
        run: |
          pdm install --group :all

      - name: Load cache data
        uses: actions/cache/restore@v4
        with:
          path: data
          key: ${{ runner.os }}-meps-reduced-example-data-v0.1.0
          restore-keys: |
            ${{ runner.os }}-meps-reduced-example-data-v0.1.0

      - name: Run tests
        run: |
          pdm run pytest

      - name: Save cache data
        uses: actions/cache/save@v4
        with:
          path: data
          key: ${{ runner.os }}-meps-reduced-example-data-v0.1.0