From 787371cfbac4429dbf8f09765d6940e7c38a245b Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Wed, 27 Sep 2023 12:26:06 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 568927297
---
 mediapipe/python/BUILD            | 1 +
 mediapipe/python/solution_base.py | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD
index 085fbc96b..b6e1ab945 100644
--- a/mediapipe/python/BUILD
+++ b/mediapipe/python/BUILD
@@ -52,6 +52,7 @@ pybind_extension(
         "@stblib//:stb_image",
         # Type registration.
         "//mediapipe/framework:basic_types_registration",
+        "//mediapipe/framework/formats:body_rig_registration",
         "//mediapipe/framework/formats:classification_registration",
         "//mediapipe/framework/formats:detection_registration",
         "//mediapipe/framework/formats:landmark_registration",
diff --git a/mediapipe/python/solution_base.py b/mediapipe/python/solution_base.py
index 020d16371..ad57ef07c 100644
--- a/mediapipe/python/solution_base.py
+++ b/mediapipe/python/solution_base.py
@@ -40,6 +40,7 @@ from mediapipe.calculators.util import landmarks_smoothing_calculator_pb2
 from mediapipe.calculators.util import logic_calculator_pb2
 from mediapipe.calculators.util import thresholding_calculator_pb2
 from mediapipe.framework import calculator_pb2
+from mediapipe.framework.formats import body_rig_pb2
 from mediapipe.framework.formats import classification_pb2
 from mediapipe.framework.formats import detection_pb2
 from mediapipe.framework.formats import landmark_pb2
@@ -192,6 +193,12 @@ NAME_TO_TYPE: Mapping[str, 'PacketDataType'] = {
         PacketDataType.PROTO_LIST,
     '::std::vector<::mediapipe::NormalizedRect>':
         PacketDataType.PROTO_LIST,
+    '::mediapipe::Joint':
+        PacketDataType.PROTO,
+    '::mediapipe::JointList':
+        PacketDataType.PROTO,
+    '::std::vector<::mediapipe::JointList>':
+        PacketDataType.PROTO_LIST,
 }