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]