From a667949b99772f074d2bf8b99decd6b80d05dab7 Mon Sep 17 00:00:00 2001 From: Kinar Date: Mon, 18 Dec 2023 02:49:20 -0800 Subject: [PATCH] Changed import style in task_info --- mediapipe/tasks/python/core/task_info.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mediapipe/tasks/python/core/task_info.py b/mediapipe/tasks/python/core/task_info.py index 5d039a034..96bb5dac9 100644 --- a/mediapipe/tasks/python/core/task_info.py +++ b/mediapipe/tasks/python/core/task_info.py @@ -21,7 +21,7 @@ from mediapipe.calculators.core import flow_limiter_calculator_pb2 from mediapipe.framework import calculator_options_pb2 from mediapipe.framework import calculator_pb2 from mediapipe.tasks.python.core.optional_dependencies import doc_controls -from google.protobuf.any_pb2 import Any +from google.protobuf import any_pb2 @doc_controls.do_not_generate_docs @@ -98,7 +98,7 @@ class TaskInfo: node_config.options.CopyFrom(task_subgraph_options) else: # Use the Any type for task_subgraph_options (proto3) - task_subgraph_options = Any() + task_subgraph_options = any_pb2.Any() task_subgraph_options.Pack(self.task_options.to_pb2()) node_config.node_options.append(task_subgraph_options)