From 1562b84613647ab493d7877df45dd8a3bf1d8674 Mon Sep 17 00:00:00 2001 From: tfx-team Date: Tue, 6 Aug 2024 11:18:35 -0700 Subject: [PATCH] 2 graphs in one request to DG Oracle PiperOrigin-RevId: 660028166 --- tfx/utils/proto_utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tfx/utils/proto_utils.py b/tfx/utils/proto_utils.py index de5abf4fd7..381b557274 100644 --- a/tfx/utils/proto_utils.py +++ b/tfx/utils/proto_utils.py @@ -95,9 +95,8 @@ def _create_proto_instance_from_name( message_name: str, pool: descriptor_pool.DescriptorPool) -> ProtoMessage: """Creates a protobuf message instance from a given message name.""" message_descriptor = pool.FindMessageTypeByName(message_name) - factory = message_factory.MessageFactory(pool) - message_type = factory.GetPrototype(message_descriptor) - return message_type() + message_class = message_factory.GetMessageClass(message_descriptor) + return message_class() def get_pool_with_descriptors(