Address comments

python · Jun 7, 2024 · c8e2982 · c8e2982
1 parent 16ef0ee
commit c8e2982
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/stdlib/email/_policybase.pyi b/stdlib/email/_policybase.pyi
@@ -18,7 +18,7 @@ class Policy(_PolicyBase, Generic[_M], metaclass=ABCMeta):
     cte_type: str
     raise_on_defect: bool
     mangle_from_: bool
-    message_factory: Callable[[Policy], Message] | None
+    message_factory: Callable[[Policy], _M] | None
     def __init__(
         self,
         *,
@@ -27,10 +27,10 @@ class Policy(_PolicyBase, Generic[_M], metaclass=ABCMeta):
         cte_type: str = "8bit",
         raise_on_defect: bool = False,
         mangle_from_: bool = False,
-        message_factory: Callable[[Policy], Message] | None = None,
+        message_factory: Callable[[Policy], _M] | None = None,
     ) -> None: ...
-    def handle_defect(self, obj: Message, defect: MessageDefect) -> None: ...
-    def register_defect(self, obj: Message, defect: MessageDefect) -> None: ...
+    def handle_defect(self, obj: _M, defect: MessageDefect) -> None: ...
+    def register_defect(self, obj: _M, defect: MessageDefect) -> None: ...
     def header_max_count(self, name: str) -> int | None: ...
     @abstractmethod
     def header_source_parse(self, sourcelines: list[str]) -> tuple[str, str]: ...