Fix Literal type labeling for Python <3.10.

matthewwardrop · Aug 15, 2024 · e82beb5 · e82beb5
1 parent a370649
commit e82beb5
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/spec_classes/utils/type_checking.py b/spec_classes/utils/type_checking.py
@@ -148,11 +148,9 @@ def type_label(attr_type: Type) -> str:
     method general.
     """
     if hasattr(attr_type, "__origin__"):  # Generics
-        label = type_label(attr_type.__origin__)
         if attr_type.__origin__ is Literal:
-            return (
-                f"{label}[{', '.join(repr(arg) for arg in attr_type.__args__)}]"
-            )
+            return f"Literal[{', '.join(repr(arg) for arg in attr_type.__args__)}]"
+        label = type_label(attr_type.__origin__)
         if hasattr(attr_type, "__args__") and not any(
             isinstance(arg, TypeVar) for arg in attr_type.__args__
         ):