Function policy_worker

codeclash/arenas/cyborg/runtime/run_cyborg.py:82–120

(
    command_queue: mp.Queue,
    result_queue: mp.Queue,
    startup_queue: mp.Queue,
    player_name: str,
    agent_name: str,
    path: str,
)

def policy_worker(
    command_queue: mp.Queue,
    result_queue: mp.Queue,
    startup_queue: mp.Queue,
    player_name: str,
    agent_name: str,
    path: str,
) -> None:
    try:
        module = load_policy_module(player_name, path)
    except BaseException as exc:
        startup_queue.put(
            {
                "error": f"{type(exc).__name__}: {exc}",
                "traceback": traceback.format_exc(limit=5),
            }
        )
        return
    startup_queue.put({"ready": True})

    while True:
        try:
            request = command_queue.get()
        except (EOFError, KeyboardInterrupt):
            return
        if request is None:
            return
        request_id = request["request_id"]
        try:
            action = module.decide(request["observation"], request["action_space"])
            result_queue.put({"request_id": request_id, "action": action})
        except BaseException as exc:
            result_queue.put(
                {
                    "request_id": request_id,
                    "error": f"{type(exc).__name__}: {exc}",
                    "traceback": traceback.format_exc(limit=5),
                }
            )


class PolicyController: