ci-gpu workflow (graphistry/pygraphistry)

name: ci-gpu
on:
  pull_request:
    types:
      - labeled
  workflow_dispatch:
  repository_dispatch:
    types:
      - trigger-ci-gpu
 
permissions:
  contents: read
 
concurrency:
  # Cancel previous runs of the same workflow/PR/ref. On non-PR events
  # (workflow_dispatch, repository_dispatch) `pull_request.labels` is null;
  # `contains(null, _)` returns false -> `!contains(...)` evaluates true,
  # so cancellation defaults on. The `skip-cancel` PR label opts out on PRs.
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: ${{ !contains(github.event.pull_request.labels.*.name, 'skip-cancel') }}
 
env:
  UV_EXCLUDE_NEWER: "6 days"
 
jobs:
 
  stub_mt:
    name: Always-succeed step to prevent appearing as a failure
    runs-on: latchkey-small
    timeout-minutes: 10
    steps:
      - name: Do nothing
        run: echo "Do nothing"
 
  gpu-disabled-guard:
    name: GPU lane temporarily disabled
    if: |
      github.event_name == 'workflow_dispatch' ||
      github.event_name == 'repository_dispatch' ||
      (github.event_name == 'pull_request' && contains(github.event.pull_request.labels.*.name, 'gpu-ci'))
    runs-on: latchkey-small
    timeout-minutes: 10
    steps:
      - name: Fail with re-enable guidance
        run: |
          echo "::error::ci-gpu is temporarily disabled while gpu_public is unavailable and PR-D hardening is pending."
          echo "::error::Re-enable only after issue #1130 acceptance criteria are satisfied."
          exit 1
 
  gpu-permission:
    timeout-minutes: 30
    # Temporarily disabled while gpu_public is unavailable and PR-D hardening is pending.
    # Re-enable only after issue #1130 criteria are met.
    if: ${{ vars.GRAPHISTRY_ENABLE_GPU_PUBLIC == 'true' }}  # see #1130; disabled by default
    runs-on: latchkey-small
    steps:
      - name: Do nothing
        run: echo "Do nothing"
 
  test-full-ai:
    timeout-minutes: 30
    # Temporarily disabled while gpu_public is unavailable and PR-D hardening is pending.
    # Re-enable only after issue #1130 criteria are met.
    if: ${{ vars.GRAPHISTRY_ENABLE_GPU_PUBLIC == 'true' }}  # see #1130; disabled by default
    runs-on: gpu_public
 
    strategy:
      matrix:
        # RAPIDS GPU images currently support <= 3.13
        python-version: ['3.10', 3.11, 3.12, '3.13']
 
    steps:
 
    - name: Checkout repo
      uses: actions/checkout@v4
      with:
        persist-credentials: false
 
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
      with:
        cache: 'pip'
        python-version: ${{ matrix.python-version }}
 
    - name: nvidia-smi
      run: nvidia-smi
 
    - name: Install test dependencies
      run: |
        python -m venv pygraphistry
        source pygraphistry/bin/activate
        python -m pip install --upgrade pip uv
        uv pip install --extra-index-url=https://pypi.nvidia.com -e .[test,testai,ai,rapids]
        echo "skrub: `pip show skrub | grep Version`"
        echo "pandas: `pip show pandas | grep Version`"
        echo "numpy: `pip show numpy | grep Version`"
        echo "scikit-learn: `pip show scikit-learn | grep Version`"
        echo "scipy: `pip show scipy | grep Version`"
        echo "umap-learn: `pip show umap-learn | grep Version`"
 
    - name: Test RAPIDS
      shell: 'script -q -e -c "bash {0}"'  # tty
      run: |
        source pygraphistry/bin/activate
        python -c 'import cudf; v = cudf.DataFrame({\"x\": [1,2,3]}).x.sum(); print(\"got\", v); assert v == 6'
 
    - name: Type check
      env:
          PYTHON_VERSION: ${{ matrix.python-version }}
      run: |
        source pygraphistry/bin/activate
        ./bin/typecheck.sh
 
    - name: Run all tests
      run: |
        source pygraphistry/bin/activate
        python -m pytest --version
        python -B -m pytest -vv -n auto --dist loadfile