Merge pull request #286 from edenai/SD2-878-google-video-person-track…

…ing-key-error [fix] google video person tracking KeyError on landmark_tracking
edenai · Dec 9, 2024 · e8c12ee · e8c12ee
2 parents f0409eb + 1aaa08d
commit e8c12ee
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/edenai_apis/apis/google/google_video_api.py b/edenai_apis/apis/google/google_video_api.py
@@ -519,10 +519,11 @@ def video__person_tracking_async__get_job_result(
                             # Landmarks
                             landmark_output = {}
                             for land in time_stamped_object.get("landmarks", []):
-                                landmark_output[land["name"]] = [
-                                    land["point"]["x"],
-                                    land["point"]["y"],
-                                ]
+                                if landmark_name := land.get("name"):
+                                    landmark_output[landmark_name] = [
+                                        land["point"]["x"],
+                                        land["point"]["y"],
+                                    ]
                             landmark_tracking = PersonLandmarks(
                                 nose=landmark_output.get("nose", []),
                                 eye_left=landmark_output.get("left_eye", []),