tests.yaml

name: Unittests

on:
  pull_request:
  push:
    branches:
      - "master"

jobs:
  unittests:
    strategy:
      matrix:
        os: ["ubuntu-latest"]
        python-version: ["3.10"]
        torch-version: ["2.0.1", "2.1.1", "2.2.1", "2.3.1", "2.4.1"]
    runs-on: ${{ matrix.os }}
    steps:
        - uses: actions/checkout@v2
        - name: Set up Python ${{ matrix.python-version }}
          uses: actions/setup-python@v2
          with:
              python-version: ${{ matrix.python-version }}
        - name: Install dependencies
          run: |
              python -m pip install --upgrade pip
              pip install "numpy==1.26.4" scikit-learn flake8 setuptools numba
        - name: Install Torch ${{ matrix.torch-version }}
          run: pip install torch==${{ matrix.torch-version }}+cpu -f https://download.pytorch.org/whl/torch
        - name: Build package
          run: |
              python setup.py build_ext --inplace
        - name: Lint with flake8
          run: |
              # stop the build if there are Python syntax errors or undefined names
              flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
              # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
              flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
        - name: Test with unittest
          run: |
              python -m unittest -v