src/scripts/test.sh

#!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "$0")/.."
ENV="${BRAINMU_ENV:-$PWD/.venv}"
[[ -x "$ENV/bin/python" ]] || { echo "Run bash scripts/setup_uv.sh first, or set BRAINMU_ENV to an existing compatible environment."; exit 2; }
export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"
export TOKENIZERS_PARALLELISM=false
exec uv run --no-project --python "$ENV/bin/python" python scripts/infer.py --expected-count 1000 "$@"