#!/bin/bash
set -e

ROUNDS="${1:-1}"
NODES="${2:-2}"

cd "$(dirname "$0")"

mkdir -p ../reports

docker compose -f docker-compose.cpu.yml run \
  -v "$(pwd)/../reports:/app/reports" \
  scheduler \
  bash -c "python -m sched run --script jobs/mnist.py --compose --trace --rounds $ROUNDS --nodes $NODES --report reports/report.html 2>&1 | tee reports/log.txt"

docker compose -f docker-compose.cpu.yml down

echo "Done. See reports/report.html and reports/log.txt"