docarray · hanxiao · Jan 26, 2022 · Jan 26, 2022 · Jan 26, 2022
diff --git a/.github/requirements-cicd.txt b/.github/requirements-cicd.txt
diff --git a/.github/requirements-test.txt b/.github/requirements-test.txt
diff --git a/.github/workflows/cd.yml b/.github/workflows/cd.yml
@@ -40,9 +40,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           python -m pip install wheel
-          pip install -r .github/requirements-test.txt
-          pip install -r .github/requirements-cicd.txt
-          pip install --no-cache-dir ".[full]"
+          pip install --no-cache-dir ".[full,test]"
       - name: Test
         id: test
         run: |

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -111,9 +111,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           python -m pip install wheel
-          pip install -r .github/requirements-test.txt
-          pip install -r .github/requirements-cicd.txt
-          pip install --no-cache-dir ".[full]"
+          pip install --no-cache-dir ".[full,test]"
       - name: Test
         id: test
         run: |

diff --git a/docs/index.md b/docs/index.md
@@ -52,6 +52,14 @@ The following dependencies will be installed to enable additional features:
 Alternatively, you can first do basic installation and then install missing dependencies on-demand. 
 ````
 
+````{tab} Developement install
+
+```bash
+pip install "docarray[full,test]"
+```
+
+This will install all requirements for reproducing tests on your local dev environment.
+````
 
 
 ```pycon

diff --git a/setup.py b/setup.py
@@ -53,7 +53,26 @@
             'av',
             'fastapi',
             'uvicorn',
-        ]
+        ],
+        'test': [
+            'pytest',
+            'pytest-timeout',
+            'pytest-mock',
+            'pytest-cov',
+            'pytest-repeat',
+            'pytest-reraise',
+            'mock',
+            'pytest-custom_exit_code',
+            'black',
+            'tensorflow==2.7.0',
+            'paddlepaddle==2.2.0',
+            'torch==1.9.0',
+            'torchvision==0.10.0',
+            'datasets',
+            'onnx',
+            'onnxruntime',
+            'jupyterlab',
+        ],
     },
     classifiers=[
         'Development Status :: 5 - Production/Stable',