Skip to content

Add dict syntax to the einops layers example #450

Add dict syntax to the einops layers example

Add dict syntax to the einops layers example #450

Workflow file for this run

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