Run CPU tests on a new dep group all-cpu (#616)

* make tests run optional deps * precommit * fix order * reprompt tests
mosaicml · Sep 20, 2023 · 299e737 · 299e737
1 parent c782e30
commit 299e737
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/pytest-cpu.yaml b/.github/workflows/pytest-cpu.yaml
@@ -27,7 +27,7 @@ jobs:
         set -ex
         export PATH=/composer-python:$PATH
         python -m pip install --upgrade 'pip<23' wheel
-        python -m pip install --upgrade .[dev]
+        python -m pip install --upgrade .[all-cpu]
     - name: Run Tests
       id: tests
       run: |

diff --git a/setup.py b/setup.py
@@ -105,6 +105,8 @@
     'openai==0.27.8',
     'tiktoken==0.4.0',
 ]
+extra_deps['all-cpu'] = set(
+    dep for key, deps in extra_deps.items() for dep in deps if 'gpu' not in key)
 extra_deps['all'] = set(dep for deps in extra_deps.values() for dep in deps)
 
 setup(