Optimize and fix prompt generation #64
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Run Smoke Test | |
on: | |
push: | |
branches: [master, main] | |
pull_request: | |
branches: [master, main] | |
workflow_dispatch: | |
env: | |
PYTHON_DEFAULT_VERSION: "3.11" | |
jobs: | |
test: | |
timeout-minutes: 10 | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v3 | |
with: | |
fetch-depth: 0 | |
- name: Set up Python ${{ env.PYTHON_DEFAULT_VERSION }} | |
uses: actions/setup-python@v4 | |
with: | |
python-version: ${{ env.PYTHON_DEFAULT_VERSION }} | |
- name: Run Test | |
run: | | |
cd src/compute_horde_prompt_gen | |
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 | |
ls output/ | |
echo -e "\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: $(cat output/prompts_uuid{$i}.txt)" | |
exit 1 | |
fi | |
done | |
echo "OK" |