add mypy to ci testing (#353)

.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/

setup.py

@@ -30,6 +30,7 @@ setup(name='tinygrad',
             "protobuf~=3.19.0",
             "onnx",
             "onnx2torch",
+            "mypy",
         ],
       },
       include_package_data=True)