From abd484a9f7048dc4a1d89b7d8ab2010079432045 Mon Sep 17 00:00:00 2001
From: wozeparrot <wozeparrot@gmail.com>
Date: Thu, 26 Sep 2024 16:44:59 +0800
Subject: [PATCH] fix: need numpy for docs and testing (#6766)

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index ba1daa8c82..92f1485af2 100644
--- a/setup.py
+++ b/setup.py
@@ -37,6 +37,7 @@ setup(name='tinygrad',
         ],
         #'mlperf': ["mlperf-logging @ git+https://github.com/mlperf/logging.git@4.0.0-rc2"],
         'testing': [
+            "numpy",
             "torch",
             "pillow",
             "pytest",
@@ -63,7 +64,8 @@ setup(name='tinygrad',
             "mkdocstrings[python]",
             "markdown-callouts",
             "markdown-exec[ansi]",
-            "black"
+            "black",
+            "numpy",
         ],
         'testing_tf': [
             "tensorflow==2.15.1",