|
|
name: CI-Import-Check |
|
|
|
|
|
on: |
|
|
push: |
|
|
pull_request: |
|
|
paths: |
|
|
- "**" |
|
|
|
|
|
jobs: |
|
|
ci-import-check: |
|
|
runs-on: ubuntu-latest |
|
|
|
|
|
|
|
|
container: |
|
|
image: pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime |
|
|
|
|
|
steps: |
|
|
- uses: actions/checkout@v2 |
|
|
|
|
|
- name: Update base dependencies |
|
|
run: | |
|
|
apt-get update && apt-get install -y build-essential |
|
|
apt-get install -y libsndfile1 make |
|
|
|
|
|
- name: Install nemo dependencies |
|
|
id: nemo-wheel |
|
|
run: | |
|
|
# install test requirements |
|
|
pip install -r requirements/requirements_test.txt |
|
|
# Build nemo as a wheel |
|
|
pip install build |
|
|
python -m build --no-isolation --wheel |
|
|
# Preserve wheel location |
|
|
DIST_FILE=$(find ./dist -name "*.whl" | head -n 1) |
|
|
echo "::set-output name=DIST_FILE::${DIST_FILE}" |
|
|
|
|
|
- name: Test ASR Domain Imports |
|
|
run: | |
|
|
# Install NeMo Domain |
|
|
pip install "${{ steps.nemo-wheel.outputs.DIST_FILE }}[asr]" |
|
|
# Run import checks |
|
|
python tests/core_ptl/check_imports.py --domain "asr" |
|
|
# Uninstall NeMo |
|
|
pip uninstall -y nemo_toolkit |
|
|
|
|
|
- name: Test TTS Domain Imports |
|
|
run: | |
|
|
# Install NeMo Domain |
|
|
pip install "${{ steps.nemo-wheel.outputs.DIST_FILE }}[tts]" |
|
|
# Run import checks |
|
|
python tests/core_ptl/check_imports.py --domain "tts" |
|
|
# Uninstall NeMo |
|
|
pip uninstall -y nemo_toolkit |
|
|
|
|
|
- name: Test NLP Domain Imports |
|
|
run: | |
|
|
# Install NeMo Domain |
|
|
pip install "${{ steps.nemo-wheel.outputs.DIST_FILE }}[nlp]" |
|
|
# Run import checks |
|
|
python tests/core_ptl/check_imports.py --domain "nlp" |
|
|
# Uninstall NeMo |
|
|
pip uninstall -y nemo_toolkit |
|
|
|
|
|
|
