From e41aff47db2416eba99295fb596d1580d8838d25 Mon Sep 17 00:00:00 2001 From: Roman Shapovalov Date: Mon, 6 Jan 2025 04:17:57 -0800 Subject: [PATCH] Adding default values to FrameData for internal usage Summary: Ensuring all fields in FrameData have defaults. Reviewed By: bottler Differential Revision: D67762780 fbshipit-source-id: b680d29a1a11689850905978df544cdb4eb7ddcd --- pytorch3d/implicitron/dataset/frame_data.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch3d/implicitron/dataset/frame_data.py b/pytorch3d/implicitron/dataset/frame_data.py index 631bea8a..9cd7e75f 100644 --- a/pytorch3d/implicitron/dataset/frame_data.py +++ b/pytorch3d/implicitron/dataset/frame_data.py @@ -122,9 +122,9 @@ class FrameData(Mapping[str, Any]): meta: A dict for storing additional frame information. """ - frame_number: Optional[torch.LongTensor] - sequence_name: Union[str, List[str]] - sequence_category: Union[str, List[str]] + frame_number: Optional[torch.LongTensor] = None + sequence_name: Union[str, List[str]] = "" + sequence_category: Union[str, List[str]] = "" frame_timestamp: Optional[torch.Tensor] = None image_size_hw: Optional[torch.LongTensor] = None effective_image_size_hw: Optional[torch.LongTensor] = None