We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0082555 commit e759b33Copy full SHA for e759b33
tfx/utils/proto_utils.py
@@ -95,9 +95,8 @@ def _create_proto_instance_from_name(
95
message_name: str, pool: descriptor_pool.DescriptorPool) -> ProtoMessage:
96
"""Creates a protobuf message instance from a given message name."""
97
message_descriptor = pool.FindMessageTypeByName(message_name)
98
- factory = message_factory.MessageFactory(pool)
99
- message_type = factory.GetPrototype(message_descriptor)
100
- return message_type()
+ message_class = message_factory.GetMessageClass(message_descriptor)
+ return message_class()
101
102
103
def get_pool_with_descriptors(
0 commit comments