chore: fix enums in stubs

to address recent mypy changes
Aiven-Open · Jan 22, 2025 · 569aa1f · 569aa1f
1 parent b5542cc
commit 569aa1f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/stubs/confluent_kafka/admin/_config.pyi b/stubs/confluent_kafka/admin/_config.pyi
@@ -1,5 +1,6 @@
 from ._resource import ResourceType
 from enum import Enum
+from typing import cast
 
 class ConfigResource:
     Type = ResourceType
@@ -12,7 +13,7 @@ class ConfigResource:
     ) -> None: ...
 
 class ConfigSource(Enum):
-    UNKNOWN_CONFIG: int
-    DYNAMIC_TOPIC_CONFIG: int
+    UNKNOWN_CONFIG = cast(int, ...)
+    DYNAMIC_TOPIC_CONFIG = cast(int, ...)
 
 class ConfigEntry: ...
diff --git a/stubs/confluent_kafka/admin/_resource.pyi b/stubs/confluent_kafka/admin/_resource.pyi
@@ -1,4 +1,5 @@
 from enum import Enum
+from typing import cast
 
 class ResourceType(Enum):
-    TOPIC: int
+    TOPIC = cast(int, ...)