adithya9903's picture
Deploy PolyGuard HF training Space
fd0c71a verified
from app.agents.orchestrator import Orchestrator
from app.env.env_core import PolyGuardEnv
def test_orchestrator_run_step() -> None:
env = PolyGuardEnv()
env.reset(seed=55, difficulty="easy")
orchestrator = Orchestrator(env)
out = orchestrator.run_step()
assert "final_action" in out
assert "reward" in out