Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/prime-rl-configs/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ dependencies = [
"renderers>=0.1.9.dev10",
"tomli>=2.2.1",
"tomli-w>=1.2.0",
"verifiers>=0.2.2.dev73",
"verifiers>=0.2.2.dev76",
]

[build-system]
Expand Down
14 changes: 7 additions & 7 deletions packages/prime-rl-configs/src/prime_rl/configs/orchestrator.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,12 +122,12 @@ def to_sampling_args(self) -> dict[str, Any]:
return args


class ServingConfig(vf.ServingConfig):
"""Verifiers' serving block with ``address`` back to optional. Verifiers defaults it
to the address its own ``serve`` CLI binds; here the question is whether to spawn a
server or connect to one already running, and that answer has to survive the
resolved config being written to a file and read back — so it must be a *value*
(``None``), not field-set metadata, which a round-trip drops."""
class ServeConfig(vf.ServeConfig):
"""Verifiers' serve block with ``address`` back to optional. Verifiers defaults it
to a fixed local bind address; here the question is whether to spawn a server or
connect to one already running, and that answer has to survive the resolved config
being written to a file and read back — so it must be a *value* (``None``), not
field-set metadata, which a round-trip drops."""

address: str | None = None
"""ZMQ address of an external env server (e.g. ``tcp://host:5000``). When set, the orchestrator connects to that server instead of spawning one; when None, it spawns a subprocess env server on a free port. ``pool`` sizes the spawned server."""
Expand All @@ -141,7 +141,7 @@ class EnvConfig(BaseConfig):
env: SerializeAsAny[vf.EnvConfig] = vf.SingleAgentEnvConfig()
"""The verifiers environment — which env, its seed taskset, each agent, its knobs. Narrowed to the selected env's config class by the env id, else the taskset id."""

serve: ServingConfig = ServingConfig()
serve: ServeConfig = ServeConfig()
"""How the env server is run: ``serve.pool`` sizes the spawned server, ``serve.address`` points at an external one instead, and ``serve.max_concurrent`` bounds one worker's episodes in flight (unset = unbounded; the dispatcher's ``max_inflight_episodes`` is the run's bound)."""

legacy: vf.LegacyEnvConfig = vf.LegacyEnvConfig()
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ dependencies = [
"setproctitle>=1.3.0",
"uvloop>=0.21.0",
"torchtitan",
"verifiers[harbor]>=0.2.2.dev73",
"verifiers[harbor]>=0.2.2.dev76",
"renderers",
"dion",
"tilelang>=0.1.8",
Expand Down
Loading
Loading