From 89b8b39d9cb98e2ca60de299befc94f4f4637bc8 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Sat, 13 May 2023 21:25:36 -0700
Subject: [PATCH] fix mypy

---
 .pre-commit-config.yaml | 2 +-
 extra/helpers.py        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 21d770a2e5..198054e917 100644
--- 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
index 8c0f1bc902..0084963ef9 100644
--- 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()