Align PolicyAgent prompt with training format#31

Merged

abrichr merged 6 commits into

mainfrom

fix/policy-agent-prompt-alignment

Feb 25, 2026

Commits on Feb 25, 2026

Align PolicyAgent prompt with training format from convert_demos.py

abrichr
and
claude
committed
Fix PolicyAgent to call predict_action_from_sample (not predict)

abrichr
and
claude
committed
Fix _action_to_string to match training format from convert_demos

abrichr
and
claude
committed
fix(modal): increase inference timeout from 300s to 600s

abrichr
and
claude
committed
fix: remove dead system prompt from PolicyAgent._build_sample()

abrichr
and
claude
committed
style: ruff format agent.py

abrichr
and
claude
committed