fix: add missing tasks arg to get_transition_events()

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Alex Cheema
2026-02-17 09:39:46 -08:00
parent 000058aece
commit de5c1212ee
+1 -2
View File
@@ -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):