name: CI-Import-Check

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

jobs:
  ci-import-check:
    runs-on: ubuntu-latest

    # Check https://hub.docker.com/r/pytorch/pytorch/tags for latest tags
    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