pre-commit + and

tinygrad · Nov 24, 2023 · 6dffac2 · 6dffac2
1 parent 6ad2ce0
commit 6dffac2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -25,7 +25,7 @@ jobs:
       run: PYTHONPATH="." python mnist.py
     - name: Install mypy + torch for testing
       run: pip install mypy torch --extra-index-url https://download.pytorch.org/whl/cpu
-    - name: Test ops / dtype / optim
+    - name: Test ops and dtype and optim
       run: |
         PYTHONPATH="." python test/test_ops.py
         PYTHONPATH="." python test/test_dtype.py

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1,9 +1,21 @@
 repos:
   - repo: local
     hooks:
-      - id: tests
-        name: tests
-        entry: env PYTHONPATH="." pytest test/
+      - id: test ops
+        name: test ops
+        entry: env PYTHONPATH="." python test/test_ops.py
+        language: system
+        always_run: true
+        pass_filenames: false
+      - id: test dtype
+        name: test dtype
+        entry: env PYTHONPATH="." python test/test_dtype.py
+        language: system
+        always_run: true
+        pass_filenames: false
+      - id: test optim
+        name: test optim
+        entry: env PYTHONPATH="." python test/test_optim.py
         language: system
         always_run: true
         pass_filenames: false