Merge pull request #91 from scaleapi/lidar-annotation-geometry-fix

Resolve cuboid yaw from geometry dict

Author: ardila

nucleus/annotation.py

@@ -287,7 +287,7 @@ def from_json(cls, payload: dict):
             label=payload.get(LABEL_KEY, 0),
             position=Point3D.from_json(geometry.get(POSITION_KEY, {})),
             dimensions=Point3D.from_json(geometry.get(DIMENSIONS_KEY, {})),
-            yaw=payload.get(YAW_KEY, 0),
+            yaw=geometry.get(YAW_KEY, 0),
             reference_id=payload.get(REFERENCE_ID_KEY, None),
             item_id=payload.get(DATASET_ITEM_ID_KEY, None),
             annotation_id=payload.get(ANNOTATION_ID_KEY, None),

nucleus/prediction.py

@@ -196,7 +196,7 @@ def from_json(cls, payload: dict):
             label=payload.get(LABEL_KEY, 0),
             position=Point3D.from_json(geometry.get(POSITION_KEY, {})),
             dimensions=Point3D.from_json(geometry.get(DIMENSIONS_KEY, {})),
-            yaw=payload.get(YAW_KEY, 0),
+            yaw=geometry.get(YAW_KEY, 0),
             reference_id=payload.get(REFERENCE_ID_KEY, None),
             item_id=payload.get(DATASET_ITEM_ID_KEY, None),
             confidence=payload.get(CONFIDENCE_KEY, None),