Skip to content

Commit a59c139

Browse files
committed
Fully async teacher loop worker handles
1 parent 48e8ed1 commit a59c139

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

verl/experimental/fully_async_policy/agent_loop/agent_loop.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,12 @@ def __init__(
8484
config: DictConfig,
8585
server_handles: list[ray.actor.ActorHandle],
8686
reward_loop_worker_handles: list[ray.actor.ActorHandle] = None,
87+
teacher_loop_worker_handles: list[ray.actor.ActorHandle] = None,
8788
):
8889
self.server_manager = FullyAsyncLLMServerManager(config, server_handles)
89-
super().__init__(config, server_handles, reward_loop_worker_handles)
90+
super().__init__(config, server_handles, reward_loop_worker_handles, teacher_loop_worker_handles)
91+
if self.distillation_enabled:
92+
raise NotImplementedError("Distillation is not implemented in FullyAsyncAgentLoopWorker yet.")
9093
# A shared cancellation event for all agent loops running on this worker.
9194
self.cancellation_event = asyncio.Event()
9295

0 commit comments

Comments
 (0)