pip.yml

name: Kymatio Pip CI

on:
    - push
    - pull_request

jobs:
    build-pip:
        runs-on: ubuntu-20.04

        strategy:
            matrix:
                python-version: ['3.8', '3.9', '3.10']
                pytorch-version: ['1.13', '2.0']
                include:
                  - python-version: '3.7'
                    pytorch-version: '1.13'
                  - python-version: '3.11'
                    pytorch-version: '2.0'
        steps:
            - uses: actions/checkout@v1
            - name: Set up Python ${{ matrix.python-version }}
              uses: actions/setup-python@v2
              with:
                  python-version: ${{ matrix.python-version }}
            - name: Install dependencies
              run: |
                  python3 -m pip install --upgrade pip
                  python3 -m pip install pytest pytest-cov

                  if [ ${{ matrix.pytorch-version }} == '1.13' ]; then
                      pip install torch==1.13.1+cpu \
                                  torchvision==0.14.1+cpu \
                                  --extra-index-url https://download.pytorch.org/whl/cpu
                  elif [ ${{ matrix.pytorch-version }} == '2.0' ]; then
                      pip install torch==2.0.0+cpu \
                                  torchvision==0.15.1+cpu \
                                  --index-url https://download.pytorch.org/whl/cpu
                  fi
                  python3 -m pip install "tensorflow>=2.0.0a"
                  python3 -m pip install scikit-learn

                  python3 -m pip install jaxlib jax

                  python3 -m pip install -r requirements.txt
                  python3 -m pip install -r requirements_optional.txt
            - name: Set up Kymatio
              run: python3 setup.py develop
            - name: Test
              run: pytest --cov=kymatio