mirror/exo
1
0
Fork 0
mirror of https://github.com/exo-explore/exo.git synced 2025-12-23 22:27:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: FrozenSet Related Bits

Browse Source
This commit is contained in:
Arbion Halili
2025-07-16 13:45:57 +01:00
parent 520b1122a3
commit 6a671908a3
1 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
shared/types/events/events.py
View File

@@ -39,14 +39,6 @@ from shared.types.worker.common import InstanceId, NodeStatus
from shared.types.worker.instances import InstanceParams, TypeOfInstance
from shared.types.worker.runners import RunnerId, RunnerStatus, RunnerStatusType
MLXEvent = Event[
frozenset(
(
EventCategoryEnum.MutatesTaskState,
EventCategoryEnum.MutatesControlPlaneState,
)
)
]
TaskEvent = Event[EventCategoryEnum.MutatesTaskState]
InstanceEvent = Event[EventCategoryEnum.MutatesInstanceState]
ControlPlaneEvent = Event[EventCategoryEnum.MutatesControlPlaneState]
@@ -163,3 +155,23 @@ class DataPlaneEdgeReplacedAtomically(Event[EventCategoryEnum.MutatesDataPlaneSt
class DataPlaneEdgeDeleted(Event[EventCategoryEnum.MutatesDataPlaneState]):
event_type: EventTypes = DataPlaneEventTypes.DataPlaneEdgeDeleted
edge_id: DataPlaneEdgeId
"""
TEST_EVENT_CATEGORIES_TYPE = FrozenSet[
Literal[
EventCategoryEnum.MutatesTaskState,
EventCategoryEnum.MutatesControlPlaneState,
]
]
TEST_EVENT_CATEGORIES = frozenset(
(
EventCategoryEnum.MutatesTaskState,
EventCategoryEnum.MutatesControlPlaneState,
)
)
class TestEvent(Event[TEST_EVENT_CATEGORIES_TYPE]):
event_category: TEST_EVENT_CATEGORIES_TYPE = TEST_EVENT_CATEGORIES
test_id: int
"""