Fix smoke test after dropping --number_of_batches

backend-developers-ltd · Sep 23, 2024 · 645b6d6 · 645b6d6
1 parent d1d56aa
commit 645b6d6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/smoke_test.yml b/.github/workflows/smoke_test.yml
@@ -28,7 +28,7 @@ jobs:
         run: | 
           cd src/compute_horde_prompt_gen
 
-          python3 run.py --output_folder_path output/ --model_name mock --number_of_batches 5 --number_of_prompts_per_batch 20 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
+          python3 run.py --output_folder_path output/ --model_name mock --number_of_prompts_per_batch 20 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
 
           echo -e "\ngenerated batches:"
           ls

diff --git a/src/compute_horde_prompt_gen/utils.py b/src/compute_horde_prompt_gen/utils.py
@@ -28,8 +28,8 @@ def parse_output(output: str) -> list[str]:
         cleaned_line = clean_line(line)
         # we skip if line is too short or too long and not ends with ?
         # in most cases it would be just first line
-        if len(cleaned_line) > 10 and len(cleaned_line) < 300 and line.endswith("?"):
-            return [line]
+        if len(cleaned_line) > 10 and len(cleaned_line) < 300 and cleaned_line.endswith("?"):
+            return [cleaned_line]
 
     return []