Skip to content

Commit e759b33

Browse files
committed
change MessageFactory class to GetMessageClass()
PiperOrigin-RevId: 662629553
1 parent 0082555 commit e759b33

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tfx/utils/proto_utils.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,8 @@ def _create_proto_instance_from_name(
9595
message_name: str, pool: descriptor_pool.DescriptorPool) -> ProtoMessage:
9696
"""Creates a protobuf message instance from a given message name."""
9797
message_descriptor = pool.FindMessageTypeByName(message_name)
98-
factory = message_factory.MessageFactory(pool)
99-
message_type = factory.GetPrototype(message_descriptor)
100-
return message_type()
98+
message_class = message_factory.GetMessageClass(message_descriptor)
99+
return message_class()
101100

102101

103102
def get_pool_with_descriptors(

0 commit comments

Comments
 (0)