fix: need numpy for docs and testing (#6766)

2026-01-10 07:28:15 -05:00 · 2024-09-26 16:44:59 +08:00
parent 2b899164c6
commit abd484a9f7
1 changed files with 3 additions and 1 deletions
--- 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",