f

.github/workflows/build_push_image.yml

@@ -18,7 +18,6 @@ env:
 jobs:
   deploy:
     timeout-minutes: 15
-    fail-fast: false
     # runs-on: ubuntu-latest
     # runs-on: ubuntu-latest-4-cores
     runs-on:
@@ -43,6 +42,7 @@ jobs:
           ls
           echo "-----"
           pdm install
+          echo "----- done"
 
       - name: Docker build and push
         run: |
@@ -52,7 +52,7 @@ jobs:
           ls
           echo "----- down model "
           pdm run download_model.py --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}"
-          echo "-----"
+          echo "----- done down model"
           echo "-----"
           echo "-----"
 

.github/workflows/smoke_test.yml

@@ -31,10 +31,18 @@ jobs:
       - name: Run Test
         run: | 
           cd src/compute_horde_prompt_gen
-          pdm run run.py --mock_model --number_of_batches 5 --number_of_prompts_per_batch 20 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
-          echo "\nGenerated batches:"
+          mkdir output/
+
+          echo "building image with no model"
+          docker build -t compute-horde-prompt-gen .
+
+          echo "running prompt generation with mock model"
+          docker run -v ./output/:/app/output/ compute-horde-prompt-gen --mock_model --number_of_batches 5 --number_of_prompts_per_batch 20 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
+
+          echo "\ngenerated batches:"
           ls output/
-          echo "\nChecking if prompts are generated"
+
+          echo "\nchecking if prompts are generated fine"
           for i in $(seq 1 5); do
             if [ $(cat output/prompts_uuid$i.txt | wc -l) -ne 20 ]; then
               echo "Missing prompts"