more tests

Signed-off-by: Oliver Koenig <okoenig@nvidia.com>
NVIDIA · Nov 28, 2024 · b3f982e · b3f982e
1 parent 81d9959
commit b3f982e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/.github/workflows/import-test.yml b/.github/workflows/import-test.yml
@@ -15,7 +15,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest]
-        collection: [asr, tts, nlp]
+        collection: [asr, nlp, tts, multimodal, audio, slu]
         python: ['3.10', '3.11', '3.12']
     steps:
     - name: Checkout repo
@@ -40,7 +40,7 @@ jobs:
         pip install -r requirements/requirements_test.txt
         
         # Install NeMo Domain
-        pip install "${{ steps.nemo-wheel.outputs.DIST_FILE }}[${{ matrix.collection }}]"
+        pip install "${{ steps.nemo-wheel.outputs.DIST_FILE }}[test,${{ matrix.collection }}]"
     
     - name: Run ${{ matrix.collection }} checks
       run: |

diff --git a/setup.py b/setup.py
@@ -84,13 +84,6 @@ def req_file(filename, folder="requirements"):
 
 # Add lightning requirements as needed
 extras_require['common'] = list(chain(extras_require['common'], extras_require['core']))
-extras_require['test'] = list(
-    chain(
-        extras_require['tts'],
-        extras_require['core'],
-        extras_require['common'],
-    )
-)
 extras_require['asr'] = list(chain(extras_require['asr'], extras_require['core'], extras_require['common']))
 extras_require['nlp'] = list(
     chain(