LSC: Remove deprecated protobuf python MessageFactory.CreatePrototype…

…() usages PiperOrigin-RevId: 683880505
pybind · Oct 9, 2024 · ac7ffc7 · ac7ffc7
1 parent ed430af
commit ac7ffc7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pybind11_protobuf/tests/dynamic_message_test.py b/pybind11_protobuf/tests/dynamic_message_test.py
@@ -19,7 +19,6 @@
 from pybind11_protobuf.tests import test_pb2
 
 POOL = descriptor_pool.DescriptorPool()
-FACTORY = message_factory.MessageFactory(POOL)
 POOL.Add(
     descriptor_pb2.FileDescriptorProto(
         name='pybind11_protobuf/tests',
@@ -42,15 +41,15 @@
 
 def get_py_dynamic_message(value=5):
   """Returns a dynamic message that is wire-compatible with IntMessage."""
-  prototype = FACTORY.CreatePrototype(
+  prototype = message_factory.GetMessageClass(
       POOL.FindMessageTypeByName('pybind11.test.DynamicMessage'))
   msg = prototype(value=value)
   return msg
 
 
 def get_py_dynamic_int_message(value=5):
   """Returns a dynamic message named pybind11.test.IntMessage."""
-  prototype = FACTORY.CreatePrototype(
+  prototype = message_factory.GetMessageClass(
       POOL.FindMessageTypeByName('pybind11.test.IntMessage'))
   msg = prototype(value=value)
   return msg