From de5c1212eec985aeedc47f75948b0f2efd7ce873 Mon Sep 17 00:00:00 2001 From: Alex Cheema Date: Tue, 17 Feb 2026 09:39:46 -0800 Subject: [PATCH] fix: add missing tasks arg to get_transition_events() Co-Authored-By: Claude Opus 4.6 --- src/exo/master/main.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/exo/master/main.py b/src/exo/master/main.py index daa9727e..90ae0539 100644 --- a/src/exo/master/main.py +++ b/src/exo/master/main.py @@ -410,7 +410,6 @@ class Master: ) generated_events.extend(transition_events) case DistributeModel(): - from exo.shared.models.model_cards import ModelCard from exo.shared.types.worker.instances import InstanceMeta from exo.shared.types.worker.shards import Sharding @@ -437,7 +436,7 @@ class Master: if instance_id not in self.state.instances: instance.shard_assignments.transfer_only = True transition_events = get_transition_events( - self.state.instances, placement + self.state.instances, placement, self.state.tasks ) generated_events.extend(transition_events) case SendInputChunk(chunk=chunk):