diff --git a/.github/workflows/api_tests.yml b/.github/workflows/api_tests.yml new file mode 100644 index 0000000000..5a3baaad9e --- /dev/null +++ b/.github/workflows/api_tests.yml @@ -0,0 +1,53 @@ +name: api_tests + +on: + push: + branches: + - main +jobs: + check_skip: + runs-on: ubuntu-latest + if: "! contains(github.event.head_commit.message, '[ci skip]')" + steps: + - run: echo "${{ github.event.head_commit.message }}" + + test: + runs-on: ubuntu-latest + strategy: + fail-fast: false + matrix: + python-version: [3.9, "3.10", "3.11"] + experimental: [false] + steps: + - uses: actions/checkout@v3 + - name: Set up Python ${{ matrix.python-version }} + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python-version }} + architecture: x64 + cache: 'pip' + cache-dependency-path: 'requirements*' + - name: check OS + run: cat /etc/os-release + - name: set ENV + run: | + export TRAINER_TELEMETRY=0 + - name: Install dependencies + run: | + sudo apt-get update + sudo apt-get install -y --no-install-recommends git make gcc + sudo apt-get install espeak-ng + make system-deps + - name: Install/upgrade Python setup deps + run: python3 -m pip install --upgrade pip setuptools wheel + - name: Replace scarf urls + run: | + sed -i 's/https:\/\/coqui.gateway.scarf.sh\//https:\/\/github.com\/coqui-ai\/TTS\/releases\/download\//g' TTS/.models.json + - name: Install TTS + run: | + python3 -m pip install .[all] + python3 setup.py egg_info + - name: Unit tests + run: make api_tests + env: + COQUI_STUDIO_TOKEN: ${{ secrets.COQUI_STUDIO_TOKEN }} diff --git a/.github/workflows/inference_tests.yml b/.github/workflows/inference_tests.yml index 47c4b241b2..d2159027b6 100644 --- a/.github/workflows/inference_tests.yml +++ b/.github/workflows/inference_tests.yml @@ -51,5 +51,3 @@ jobs: python3 setup.py egg_info - name: Unit tests run: make inference_tests - env: - COQUI_STUDIO_TOKEN: ${{ secrets.COQUI_STUDIO_TOKEN }} diff --git a/Makefile b/Makefile index 7adea3a1fe..9f2008dc67 100644 --- a/Makefile +++ b/Makefile @@ -29,6 +29,9 @@ test_zoo: ## run zoo tests. inference_tests: ## run inference tests. nose2 -F -v -B --with-coverage --coverage TTS tests.inference_tests +api_tests: ## run api tests. + nose2 -F -v -B --with-coverage --coverage TTS tests.api_tests + data_tests: ## run data tests. nose2 -F -v -B --with-coverage --coverage TTS tests.data_tests diff --git a/TTS/tts/layers/bark/inference_funcs.py b/TTS/tts/layers/bark/inference_funcs.py index 3a73e6d4b4..3a6875ef39 100644 --- a/TTS/tts/layers/bark/inference_funcs.py +++ b/TTS/tts/layers/bark/inference_funcs.py @@ -136,7 +136,9 @@ def generate_voice( hubert_model = CustomHubert(checkpoint_path=model.config.LOCAL_MODEL_PATHS["hubert"]).to(model.device) # Load the CustomTokenizer model - tokenizer = HubertTokenizer.load_from_checkpoint(model.config.LOCAL_MODEL_PATHS["hubert_tokenizer"], map_location=model.device) + tokenizer = HubertTokenizer.load_from_checkpoint( + model.config.LOCAL_MODEL_PATHS["hubert_tokenizer"], map_location=model.device + ) # semantic_tokens = model.text_to_semantic( # text, max_gen_duration_s=seconds, top_k=50, top_p=0.95, temp=0.7 # ) # not 100% diff --git a/tests/api_tests/__init__.py b/tests/api_tests/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/tests/inference_tests/test_python_api.py b/tests/api_tests/test_python_api.py similarity index 100% rename from tests/inference_tests/test_python_api.py rename to tests/api_tests/test_python_api.py