Parameterise stage

guardian · Jan 31, 2025 · d412ebe · d412ebe
1 parent 7249ed8
commit d412ebe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/roles/whisperx/defaults/main.yml b/roles/whisperx/defaults/main.yml
@@ -1,2 +1,3 @@
 ---
 amigo_data_bucket: amigo-data-prod
+model_script_stage: PROD
diff --git a/roles/whisperx/tasks/main.yml b/roles/whisperx/tasks/main.yml
@@ -20,7 +20,7 @@
 
 - name: Download models script
   shell: |
-    aws --quiet s3 cp s3://{{ amigo_data_bucket }}/deploy/PROD/whisperx-model-fetch/download_whisperx_models.py /tmp/download_whisperx_models.py
+    aws --quiet s3 cp s3://{{ amigo_data_bucket }}/deploy/{{ model_script_stage }}/whisperx-model-fetch/download_whisperx_models.py /tmp/download_whisperx_models.py
     exit 0
 
 - name: Download whisperx models