fix mypy

ZoltanVeres · May 14, 2023 · 89b8b39 · 89b8b39
1 parent e0b2035
commit 89b8b39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -15,7 +15,7 @@ repos:
         pass_filenames: false
       - id: mypy
         name: mypy
-        entry: mypy tinygrad/ --check-untyped-defs --explicit-package-bases --warn-unreachable # --warn-return-any
+        entry: mypy tinygrad/ extra/helpers.py --check-untyped-defs --explicit-package-bases --warn-unreachable # --warn-return-any
         language: system
         always_run: true
         pass_filenames: false

diff --git a/extra/helpers.py b/extra/helpers.py
@@ -30,7 +30,7 @@ def proc(itermaker, q):
 def cross_process(itermaker, maxsize=8):
   # TODO: use cloudpickle for itermaker
   import multiprocessing
-  q = multiprocessing.Queue(maxsize)
+  q: multiprocessing.Queue = multiprocessing.Queue(maxsize)
   p = multiprocessing.Process(target=proc, args=(itermaker, q))
   p.daemon = True
   p.start()