Skip to content

Commit 1f356b3

Browse files
Merge pull request #1154 from Saga4/codeflash/optimize-from_keypoints_detection_response-m7nuxr9h_fixed
⚡️ Speed up function `from_keypoints_detection_response` by 7%
2 parents 1a32e42 + 05ba15f commit 1f356b3

1 file changed

Lines changed: 5 additions & 7 deletions

File tree

inference/core/cache/serializers.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -118,10 +118,8 @@ def from_instance_segmentation_response(
118118

119119

120120
def from_keypoints_detection_response(response: KeypointsDetectionInferenceResponse):
121-
predictions = []
122-
for pred in response.predictions:
123-
for keypoint in pred.keypoints:
124-
predictions.append(
125-
{"class": keypoint.class_name, "confidence": keypoint.confidence}
126-
)
127-
return predictions
121+
return [
122+
{"class": keypoint.class_name, "confidence": keypoint.confidence}
123+
for pred in response.predictions
124+
for keypoint in pred.keypoints
125+
]

0 commit comments

Comments
 (0)