Use TypeVar for Schema.schema type

alecthomas · Jun 25, 2024 · d522368 · d522368
1 parent 66df7be
commit d522368
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/voluptuous/schema_builder.py b/voluptuous/schema_builder.py
@@ -162,8 +162,10 @@ def Extra(_) -> None:
 ]
 # fmt: on
 
+_SchemableT = typing.TypeVar("_SchemableT", bound=Schemable)
 
-class Schema(object):
+
+class Schema(typing.Generic[_SchemableT]):
     """A validation schema.
 
     The schema is a Python tree-like structure where nodes are pattern
@@ -192,7 +194,7 @@ class Schema(object):
     }
 
     def __init__(
-        self, schema: Schemable, required: bool = False, extra: int = PREVENT_EXTRA
+        self, schema: _SchemableT, required: bool = False, extra: int = PREVENT_EXTRA
     ) -> None:
         """Create a new Schema.