python.yml

name: Python

on:
  pull_request:

jobs:
  build_and_test:
    name: Check everything builds & tests
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        os: [ubuntu-latest, macos-latest, windows-latest]
        # Lowest and highest, no version specified so that 
        # new releases get automatically tested against
        version: [{torch: torch==1.10, python: "3.8"}, {torch: torch, python: "3.10"}]
    defaults:
      run:
        working-directory: ./bindings/python
    steps:
      - name: Checkout repository
        uses: actions/checkout@v3
        

      - name: Install Rust
        uses: actions-rs/toolchain@v1
        with:
          toolchain: stable
          components: rustfmt, clippy

      - uses: Swatinem/rust-cache@v2
        with:
          workspaces: "bindings/python"

      - name: Install Python
        uses: actions/setup-python@v2
        with:
          python-version: ${{ matrix.version.python }}
          architecture: "x64"

      - name: Lint with RustFmt
        run: cargo fmt -- --check

      - name: Lint with Clippy
        run: cargo clippy --all-targets --all-features -- -D warnings

      - name: Run Audit
        run: cargo audit -D warnings

      - name: Install
        run: |
          pip install -U pip
          pip install .[numpy,tensorflow,paddlepaddle]
          pip install ${{ matrix.version.torch }}

      - name: Install (jax, flax)
        if: matrix.os != 'windows-latest'
        run: |
          pip install .[jax]
        shell: bash

      - name: Check style
        run: |
          pip install .[quality]
          black --check --line-length 119 --target-version py35 py_src/safetensors tests

      - name: Run tests
        run: |
          cargo test
          pip install .[testing]
          pytest -sv tests/