1 files changed, 2 insertions, 3 deletions
diff --git a/generator.py b/generator.py
index 2366416..3eb997f 100755
--- a/generator.py
+++ b/generator.py
@@ -1689,10 +1689,9 @@ def buildWrappers(module: str) -> None:
     #
     # Generate enum constants
     #
-    def enumsSortKey(data):
-        value = data[1]
+    def enumsSortKey(data: Tuple[str, EnumValue]) -> Tuple[Union[int, float], str]:
         try:
-            value = int(value)
+            value = int(data[1])  # type: Union[int, float]
         except ValueError:
             value = float('inf')
         return value, data[0]