diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 8652c876b4..0e7ebbd5b7 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -94,3 +94,21 @@ jobs: run: pip install -e '.[gpu,testing]' - name: Run Pytest run: GPU=1 python -m pytest -s -v + + testmypy: + name: Mypy Tests + runs-on: ubuntu-latest + + steps: + - name: Checkout Code + uses: actions/checkout@v2 + - name: Update packages + run: sudo apt-get update + - name: Set up Python 3.8 + uses: actions/setup-python@v2 + with: + python-version: 3.8 + - name: Install Dependencies + run: pip install -e '.[testing]' + - name: Run mypy + run: mypy tinygrad/ diff --git a/setup.py b/setup.py index 7440821316..624ac45536 100644 --- a/setup.py +++ b/setup.py @@ -30,6 +30,7 @@ setup(name='tinygrad', "protobuf~=3.19.0", "onnx", "onnx2torch", + "mypy", ], }, include_package_data=True)