name: Tests

on:
  push:
    branches: [main]
  pull_request:
    branches: [main]

jobs:
  test:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4

      - name: Set up Python 3.10
        uses: actions/setup-python@v5
        with:
          python-version: "3.10"

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          pip install numba==0.58.1
          pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cpu
          pip install torch_geometric
          pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.0+cpu.html
          pip install xformers==0.0.29.post1 --no-deps
          pip install pytorch-lightning yacs torchmetrics performer-pytorch tensorboardX ogb wandb seaborn dgl
          pip install scipy pandas scikit-learn matplotlib tqdm PyYAML
          pip install awkward0 uproot awkward vector lz4 xxhash tables tensorboard plotly
          pip install fastjet gradio huggingface_hub hdbscan ruff
          pip install --no-deps git+https://github.com/gregorkrz/lorentz-gatr.git
          pip install pytest

      - name: Lint with ruff
        run: |
          ruff check --select E9,F63,F7 --output-format=github tests/

      - name: Run tests
        run: |
          pytest tests/ -v --tb=short