From 7e497f052df7398262ced89c93f4e8bd573698f7 Mon Sep 17 00:00:00 2001
From: Mathieu Pichette <95452742+mpichette-apple@users.noreply.github.com>
Date: Sun, 6 Aug 2023 21:14:35 -0700
Subject: [PATCH] Add missing stubs in google.protobuf.message_factory module
 (#10537)

---
 stubs/protobuf/google/protobuf/message_factory.pyi | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/stubs/protobuf/google/protobuf/message_factory.pyi b/stubs/protobuf/google/protobuf/message_factory.pyi
index 4368ce132..e9009393a 100644
--- a/stubs/protobuf/google/protobuf/message_factory.pyi
+++ b/stubs/protobuf/google/protobuf/message_factory.pyi
@@ -12,4 +12,6 @@ class MessageFactory:
     def GetPrototype(self, descriptor: Descriptor) -> type[Message]: ...
     def GetMessages(self, files: Iterable[str]) -> dict[str, type[Message]]: ...
 
+def GetMessageClass(descriptor: Descriptor) -> type[Message]: ...
+def GetMessageClassesForFiles(files: Iterable[str], pool: DescriptorPool) -> dict[str, type[Message]]: ...
 def GetMessages(file_protos: Iterable[FileDescriptorProto], pool: DescriptorPool | None = ...) -> dict[str, type[Message]]: ...