From 9c92c35adf8f0b294cd2fb7b4a44231f4116cbef Mon Sep 17 00:00:00 2001
From: Enno Hermann <enno.hermann@idiap.ch>
Date: Wed, 20 Nov 2024 15:45:38 +0100
Subject: [PATCH] ci: allow testing out trainer/coqpit branches before release

---
 .github/workflows/integration-tests.yml | 18 ++++++++++++++++++
 .github/workflows/tests.yml             | 18 ++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
index 7af0836248..4dc8c76c1a 100644
--- a/.github/workflows/integration-tests.yml
+++ b/.github/workflows/integration-tests.yml
@@ -6,6 +6,16 @@ on:
       - main
   pull_request:
     types: [opened, synchronize, reopened]
+  workflow_dispatch:
+    inputs:
+      trainer_branch:
+        description: "Branch of Trainer to test"
+        required: false
+        default: "main"
+      coqpit_branch:
+        description: "Branch of Coqpit to test"
+        required: false
+        default: "main"
 jobs:
   test:
     runs-on: ubuntu-latest
@@ -30,6 +40,14 @@ jobs:
           sudo apt-get update
           sudo apt-get install -y --no-install-recommends git make gcc
           make system-deps
+      - name: Install custom Trainer and/or Coqpit if requested
+        run: |
+          if [[ -n "${{ github.event.inputs.trainer_branch }}" ]]; then
+            uv add git+https://github.com/idiap/coqui-ai-Trainer --branch ${{ github.event.inputs.trainer_branch }}
+          fi
+          if [[ -n "${{ github.event.inputs.coqpit_branch }}" ]]; then
+            uv add git+https://github.com/idiap/coqui-ai-coqpit --branch ${{ github.event.inputs.coqpit_branch }}
+          fi
       - name: Integration tests
         run: |
           resolution=highest
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 16b680a93c..576de150fd 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -6,6 +6,16 @@ on:
       - main
   pull_request:
     types: [opened, synchronize, reopened]
+  workflow_dispatch:
+    inputs:
+      trainer_branch:
+        description: "Branch of Trainer to test"
+        required: false
+        default: "main"
+      coqpit_branch:
+        description: "Branch of Coqpit to test"
+        required: false
+        default: "main"
 jobs:
   test:
     runs-on: ubuntu-latest
@@ -30,6 +40,14 @@ jobs:
           sudo apt-get update
           sudo apt-get install -y --no-install-recommends git make gcc
           make system-deps
+      - name: Install custom Trainer and/or Coqpit if requested
+        run: |
+          if [[ -n "${{ github.event.inputs.trainer_branch }}" ]]; then
+            uv add git+https://github.com/idiap/coqui-ai-Trainer --branch ${{ github.event.inputs.trainer_branch }}
+          fi
+          if [[ -n "${{ github.event.inputs.coqpit_branch }}" ]]; then
+            uv add git+https://github.com/idiap/coqui-ai-coqpit --branch ${{ github.event.inputs.coqpit_branch }}
+          fi
       - name: Unit tests
         run: |
           resolution=highest