Skip to content

Commit 91c6671

Browse files
ravwojdylaclaude
andcommitted
zephyr: allow to specify coordinator resources (#4005)
Per title, bump default memory to 5g. --------- Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 05d9080 commit 91c6671

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

lib/zephyr/src/zephyr/execution.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1259,6 +1259,9 @@ class ZephyrContext:
12591259
min(max_workers, num_shards), computed at first execute(). If None,
12601260
defaults to os.cpu_count() for LocalClient, or 128 for distributed clients.
12611261
resources: Resource config per worker.
1262+
coordinator_resources: Resource config for the coordinator job. The coordinator
1263+
accumulates scatter manifests for all shards in memory; increase ram for
1264+
large pipelines (many shards). Defaults to 5 GB.
12621265
chunk_storage_prefix: Storage prefix for intermediate chunks. If None, defaults
12631266
to MARIN_PREFIX/tmp/zephyr or /tmp/zephyr.
12641267
name: Descriptive name for this context, used in actor group names for debugging.
@@ -1273,6 +1276,7 @@ class ZephyrContext:
12731276
client: Client | None = None
12741277
max_workers: int | None = None
12751278
resources: ResourceConfig = field(default_factory=lambda: ResourceConfig(cpu=1, ram="1g"))
1279+
coordinator_resources: ResourceConfig = field(default_factory=lambda: ResourceConfig(cpu=1, ram="5g"))
12761280
chunk_storage_prefix: str | None = None
12771281
name: str = ""
12781282
no_workers_timeout: float | None = None
@@ -1409,7 +1413,7 @@ def execute(
14091413
_run_coordinator_job,
14101414
args=(config_path, result_path),
14111415
),
1412-
resources=ResourceConfig(cpu=1, ram="1g"),
1416+
resources=self.coordinator_resources,
14131417
)
14141418
)
14151419

0 commit comments

Comments
 (0)