test_trajdl_main.yml

name: Test TrajDL on branch main

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

jobs:
  build:

    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.10", "3.11", "3.12"]

    steps:
    - uses: actions/checkout@v4

    - name: Install build-essential and make
      run: |
        sudo apt-get update
        sudo apt-get install -y build-essential ca-certificates curl libffi-dev libgdbm-dev libncurses5-dev libnss3-dev libreadline-dev libssl-dev make time wget zlib1g-dev

    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
      with:
        python-version: ${{ matrix.python-version }}
        cache: 'pip'

    - name: Install PyTorch and dependencies from pyproject.toml
      run: |
        python -m pip install --upgrade pip
        python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
        python -m pip install .[dev,docs]

    - name: Cache datasets
      id: cache-datasets
      uses: actions/cache@v3
      with:
        path: ~/.cache/trajdl
        key: trajdl-cache-datasets

    - name: Test with pytest
      run: |
        autoflake -j 0 --remove-all-unused-imports --in-place --recursive --ignore-init-module-imports .
        isort -j -1 .
        black .
        pytest -s
    
    - name: Build package
      run: |
        if [ "${{ matrix.python-version }}" = "3.12" ]; then
          python -m build
        fi

    - name: Upload artifacts
      if: matrix.python-version == '3.12'
      uses: actions/upload-artifact@v4
      with:
        name: dist-${{ github.ref_name }}-${{ github.sha }}-${{ matrix.python-version }}
        path: dist/
        retention-days: 1
        overwrite: true