Skip to content

Commit 21d9add

Browse files
Drew KaulDrew Kaul
authored andcommitted
fix item_id in CuboidPrediction
1 parent 8f74ade commit 21d9add

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nucleus/prediction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def from_json(cls, payload: dict):
198198
dimensions=Point3D.from_json(geometry.get(DIMENSIONS_KEY, {})),
199199
yaw=payload.get(YAW_KEY, 0),
200200
reference_id=payload.get(REFERENCE_ID_KEY, None),
201-
item_id=payload.get(DATASET_ITEM_ID_KEY, None),
201+
item_id=payload.get(ITEM_ID_KEY, None),
202202
confidence=payload.get(CONFIDENCE_KEY, None),
203203
annotation_id=payload.get(ANNOTATION_ID_KEY, None),
204204
metadata=payload.get(METADATA_KEY, {}),

0 commit comments

Comments
 (0)