mention that cupy and jax now support array api (#348) #456
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Run tests | |
| on: [push, pull_request] | |
| jobs: | |
| build: | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| python-version: ['3.8', '3.10', '3.12'] | |
| # currently there is conflict between tf, oneflow and paddle in protobuf versions. | |
| # cupy is not tested because it demands gpu | |
| # oneflow testing is dropped, see details at https://github.com/Oneflow-Inc/oneflow/issues/10340 | |
| # paddle was switched off because of divergence with numpy in py3.10, paddle==2.6.1 | |
| frameworks: ['numpy pytorch tensorflow jax'] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| cache: pip | |
| - name: Check for ruff compliance | |
| run: | | |
| pip install ruff==0.6.5 && ruff check . && ruff format . --check | |
| - name: Run tests | |
| run: | | |
| pip install -e . && python -m einops.tests.run_tests ${{ matrix.frameworks }} --pip-install |