python_tests.yaml

name: Python Unit Tests

on:
  push:
    branches:
      - main
      - develop
  pull_request:
    branches:
      - main
      - develop

jobs:
  build:
    runs-on: ubuntu-latest

    steps:
    - name: Checkout repository
      uses: actions/checkout@v4

    - name: Set up Python
      uses: actions/setup-python@v5
      with:
        python-version: '3.11'
        cache: 'pip'

    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip

        # 1. Install CPU-only PyTorch first (Prevents 3GB+ download)
        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

        # 2. Install the rest of the requirements
        # (Make sure you removed 'torch' from requirements.txt!)
        pip install -r requirements.txt

        # 3. Install explicit tools
        pip install tqdm

    - name: Run Unit Tests
      run: |
        python -m unittest discover tests