From 72f08c6878f5ada7573fbd802638f52019f14f4b Mon Sep 17 00:00:00 2001 From: kinaryml Date: Tue, 8 Nov 2022 10:57:12 -0800 Subject: [PATCH] Used loops instead of list comprehension --- .../containers/classification_result.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/mediapipe/tasks/python/components/containers/classification_result.py b/mediapipe/tasks/python/components/containers/classification_result.py index 20dc6991a..aee4f0655 100644 --- a/mediapipe/tasks/python/components/containers/classification_result.py +++ b/mediapipe/tasks/python/components/containers/classification_result.py @@ -48,15 +48,13 @@ class Classifications: def to_pb2(self) -> _ClassificationsProto: """Generates a Classifications protobuf object.""" classification_list_proto = _ClassificationListProto() - classification_list_proto.classification.extend([ - _ClassificationProto( - index=category.index, - score=category.score, - label=category.category_name, - display_name=category.display_name - ) - for category in self.categories - ]) + for category in self.categories: + classification_proto = _ClassificationProto( + index=category.index, + score=category.score, + label=category.category_name, + display_name=category.display_name) + classification_list_proto.classification.append(classification_proto) return _ClassificationsProto( classification_list=classification_list_proto, head_index=self.head_index,