import-test.yml

name: CI-Import-Check

on:
  push:
  pull_request:
    paths:
      - "**"

# Check https://hub.docker.com/r/pytorch/pytorch/tags for latest tags
jobs:

  test-asr-imports:
    runs-on: ubuntu-latest
    container:
      image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
    steps:
    - name: Checkout repo
      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:  |
        pip install Cython
        # 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
  test-tts-imports:
    runs-on: ubuntu-latest
    container:
      image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
    steps:
    - name: Checkout repo
      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:  |
        pip install Cython
        # 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 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