#!/usr/bin/env bash
set -euo pipefail

uv run pytest -v ./tests/test_grpo.py --junitxml=test_results.xml || true
echo "Done running tests"

# Set the name of the output zip file
output_file="code.zip"
rm "$output_file" || true

# Compress all files in the current directory into a single zip file
zip -r "$output_file" . \
    -x '*egg-info*' \
    -x '*mypy_cache*' \
    -x '*pytest_cache*' \
    -x '*build*' \
    -x '*ipynb_checkpoints*' \
    -x '*__pycache__*' \
    -x '*.pkl' \
    -x '*.pickle' \
    -x '*.txt' \
    -x '*.log' \
    -x '*.json' \
    -x '*.out' \
    -x '*.err' \
    -x '.git*' \
    -x '.venv/*' \
    -x '*.bin' \
    -x '*.pt' \
    -x '*.pth' \
    -x '*.safetensors' \
    -x ./data/\*

echo "All files have been compressed into $output_file"