From e8d771baf3d37b044b76fe1e8ff286317c1319e2 Mon Sep 17 00:00:00 2001
From: kinaryml <kinar.ravishankar@ymedialabs.com>
Date: Tue, 8 Nov 2022 10:37:37 -0800
Subject: [PATCH] Updated landmark_detection_result

---
 .../containers/landmark_detection_result.py   | 46 +++++++++----------
 1 file changed, 23 insertions(+), 23 deletions(-)

diff --git a/mediapipe/tasks/python/components/containers/landmark_detection_result.py b/mediapipe/tasks/python/components/containers/landmark_detection_result.py
index 317290811..5e68efda1 100644
--- a/mediapipe/tasks/python/components/containers/landmark_detection_result.py
+++ b/mediapipe/tasks/python/components/containers/landmark_detection_result.py
@@ -54,23 +54,19 @@ class LandmarksDetectionResult:
     """Generates a LandmarksDetectionResult protobuf object."""
 
     landmarks = _NormalizedLandmarkListProto()
-    landmarks.landmark.extend([
-      landmark.to_pb2() for landmark in self.landmarks
-    ])
-
     classifications = _ClassificationListProto()
-    classifications.classification.extend([
-      _ClassificationProto(
-        index=category.index,
-        score=category.score,
-        label=category.category_name,
-        display_name=category.display_name) for category in self.categories
-    ])
-
     world_landmarks = _LandmarkListProto()
-    world_landmarks.landmark.extend([
-      world_landmark.to_pb2() for world_landmark in self.world_landmarks
-    ])
+
+    for landmark in self.landmarks:
+      landmarks.landmark.append(landmark.to_pb2())
+
+    for category in self.categories:
+      classifications.classification.append(
+          _ClassificationProto(
+              index=category.index,
+              score=category.score,
+              label=category.category_name,
+              display_name=category.display_name))
 
     return _LandmarksDetectionResultProto(
         landmarks=landmarks,
@@ -86,6 +82,9 @@ class LandmarksDetectionResult:
     """Creates a `LandmarksDetectionResult` object from the given protobuf object.
     """
     categories = []
+    landmarks = []
+    world_landmarks = []
+
     for classification in pb2_obj.classifications.classification:
       categories.append(
           category_module.Category(
@@ -93,14 +92,15 @@ class LandmarksDetectionResult:
               index=classification.index,
               category_name=classification.label,
               display_name=classification.display_name))
+
+    for landmark in pb2_obj.landmarks.landmark:
+      landmarks.append(_NormalizedLandmark.create_from_pb2(landmark))
+
+    for landmark in pb2_obj.world_landmarks.landmark:
+      world_landmarks.append(_Landmark.create_from_pb2(landmark)
+                             )
     return LandmarksDetectionResult(
-        landmarks=[
-            _NormalizedLandmark.create_from_pb2(landmark)
-            for landmark in pb2_obj.landmarks.landmark
-        ],
+        landmarks=landmarks,
         categories=categories,
-        world_landmarks=[
-            _Landmark.create_from_pb2(landmark)
-            for landmark in pb2_obj.world_landmarks.landmark
-        ],
+        world_landmarks=world_landmarks,
         rect=_NormalizedRect.create_from_pb2(pb2_obj.rect))