cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changed import style in task_info

Browse Source
This commit is contained in:
Kinar 2023-12-18 02:49:20 -08:00
parent 30e6b766d4
commit a667949b99
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/tasks/python/core/task_info.py
View File

@ -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_options_pb2
from mediapipe.framework import calculator_pb2 from mediapipe.framework import calculator_pb2
from mediapipe.tasks.python.core.optional_dependencies import doc_controls 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 @doc_controls.do_not_generate_docs
@ -98,7 +98,7 @@ class TaskInfo:
node_config.options.CopyFrom(task_subgraph_options) node_config.options.CopyFrom(task_subgraph_options)
else: else:
# Use the Any type for task_subgraph_options (proto3) # 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()) task_subgraph_options.Pack(self.task_options.to_pb2())
node_config.node_options.append(task_subgraph_options) node_config.node_options.append(task_subgraph_options)