fix: FrozenSet Related Bits

6a671908a3e5449a25dce9151a798c90a55c6dfc · 2025-07-16 13:45:57 +0100 · Arbion Halili

Files touched

M shared/types/events/events.py

Diff

commit 6a671908a3e5449a25dce9151a798c90a55c6dfc
Author: Arbion Halili <99731180+ToxicPine@users.noreply.github.com>
Date:   Wed Jul 16 13:45:57 2025 +0100

    fix: FrozenSet Related Bits
---
 shared/types/events/events.py | 28 ++++++++++++++++++++--------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/shared/types/events/events.py b/shared/types/events/events.py
index aabd081b..331ad87f 100644
--- a/shared/types/events/events.py
+++ b/shared/types/events/events.py
@@ -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
+"""
\ No newline at end of file

← 520b1122 fix: Many Fixes · back to Exo · fix: Fix incorrect logic e2a79350 →