action.yml

name: Setup uv environment
description: Restore uv and venv caches, and run uv sync on cache miss
inputs:
  uv-cache-key-prefix:
    description: Prefix for uv package cache key
    required: true
  venv-cache-key-prefix:
    description: Prefix for virtual environment cache key
    required: true
  cache-key-suffix:
    description: Deterministic suffix appended to cache key prefixes
    required: true
outputs:
  uv_cache_hit:
    description: Whether uv package cache had an exact key hit
    value: ${{ steps.restore-uv-cache.outputs.cache-hit }}
  venv_cache_hit:
    description: Whether venv cache had an exact key hit
    value: ${{ steps.restore-venv-cache.outputs.cache-hit }}
runs:
  using: composite
  steps:
    - name: Restore uv package cache
      id: restore-uv-cache
      uses: actions/cache/restore@v4
      with:
        path: ~/.cache/uv
        key: ${{ inputs.uv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}
        fail-on-cache-miss: false
        restore-keys: |
          ${{ inputs.uv-cache-key-prefix }}-

    - name: Restore venv cache
      id: restore-venv-cache
      uses: actions/cache/restore@v4
      with:
        path: .venv
        key: ${{ inputs.venv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}
        fail-on-cache-miss: false
        restore-keys: |
          ${{ inputs.venv-cache-key-prefix }}-

    - name: Debug cache and environment context
      shell: bash
      run: |
        set -euo pipefail
        echo "::group::setup-uv-env debug context"
        echo "uv cache key: ${{ inputs.uv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}"
        echo "venv cache key: ${{ inputs.venv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}"
        echo "uv cache exact hit: ${{ steps.restore-uv-cache.outputs.cache-hit }}"
        echo "venv cache exact hit: ${{ steps.restore-venv-cache.outputs.cache-hit }}"
        echo "workspace: $GITHUB_WORKSPACE"
        df -h
        echo "::endgroup::"

    - name: Install dependencies with uv (dev + cu12)
      if: steps.restore-venv-cache.outputs.cache-hit != 'true'
      shell: bash
      run: |
        set -euo pipefail
        export UV_LINK_MODE=copy
        echo "::group::uv sync (dev + cu12)"
        uv sync \
          --frozen \
          --group dev \
          --extra cu12
        echo "::endgroup::"
        uv run python -c "import torch; print(f'torch={torch.__version__} cuda={torch.version.cuda}')"