From fe17aaf9f8aee749ba752f1362c400076b499fed Mon Sep 17 00:00:00 2001 From: Arbion Halili <99731180+ToxicPine@users.noreply.github.com> Date: Mon, 7 Jul 2025 20:22:00 +0100 Subject: [PATCH] fix: Make master hold a queue of task data --- shared/types/states/master.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/shared/types/states/master.py b/shared/types/states/master.py index 09a5d584..51659334 100644 --- a/shared/types/states/master.py +++ b/shared/types/states/master.py @@ -24,6 +24,7 @@ from shared.types.networking.topology import ( ) from shared.types.profiling.common import NodePerformanceProfile from shared.types.states.shared import SharedState +from shared.types.tasks.common import TaskData, TaskType from shared.types.worker.instances import InstanceData, InstanceId @@ -64,8 +65,8 @@ class ControlPlaneNetworkState(State[EventCategories.ControlPlaneEventTypes]): class MasterState(SharedState): data_plane_network_state: DataPlaneNetworkState control_plane_network_state: ControlPlaneNetworkState - job_inbox: Queue[ExternalCommand] - job_outbox: Queue[ExternalCommand] + job_inbox: Queue[TaskData[TaskType]] + job_outbox: Queue[TaskData[TaskType]] cache_policy: CachePolicy[CachePolicyType]