From 6eb0e6a6501644a410d5b90171c41fd0101d2778 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Mon, 6 Feb 2023 09:27:01 -0600
Subject: [PATCH] shuffle deps: always tqdm, make linting category

---
 .github/workflows/test.yml |  2 +-
 setup.py                   | 14 ++++++++------
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 930e1e43ec..219ea3d66b 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -32,7 +32,7 @@ jobs:
       with:
         python-version: 3.8
     - name: Install dependencies
-      run: pip install -e '.[testing]'
+      run: pip install -e '.[linting,testing]'
     - name: Lint with pylint
       run: python -m pylint --disable=all -e W0311 --jobs=0 --indent-string='  ' **/*.py
     - name: Lint with flake8
diff --git a/setup.py b/setup.py
index c18dbc1c2c..f645274278 100644
--- a/setup.py
+++ b/setup.py
@@ -19,23 +19,25 @@ setup(name='tinygrad',
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License"
       ],
-      install_requires=['numpy', 'requests', 'pillow', 'networkx'],
+      install_requires=['numpy', 'requests', 'pillow', 'tqdm', 'networkx'],
       python_requires='>=3.8',
       extras_require={
         'gpu': ["pyopencl", "six"],
         'llvm': ["llvmlite"],
         'cuda': ["pycuda"],
         'triton': ["triton>=2.0.0.dev20221202"],
-        'testing': [
+        'linting': [
+            "flake8",
             "pytest",
+            "pylint",
+            "mypy",
+            "pre-commit",
+        ],
+        'testing': [
             "torch~=1.13.0",
-            "tqdm",
             "protobuf~=3.19.0",
             "onnx",
             "onnx2torch",
-            "mypy",
-            "pylint",
-            "pre-commit",
         ],
       },
       include_package_data=True)