Skip to content

Commit 3a1d45b

Browse files
Drew KaulDrew Kaul
authored andcommitted
add repr's to Scene and Frame classes
1 parent ed585c9 commit 3a1d45b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

nucleus/scene.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@ def __post_init__(self):
2727
value, DatasetItem
2828
), "All values must be DatasetItems"
2929

30+
def __repr__(self) -> str:
31+
return f"Frame(index={self.index}, items={self.items})"
32+
3033
def add_item(self, item: DatasetItem, sensor_name: str):
3134
self.items[sensor_name] = item
3235

@@ -210,6 +213,9 @@ def to_json(self) -> str:
210213

211214
@dataclass
212215
class LidarScene(Scene):
216+
def __repr__(self) -> str:
217+
return f"LidarScene(reference_id='{self.reference_id}', frames={self.get_frames()}, metadata={self.metadata})"
218+
213219
def validate(self):
214220
super().validate()
215221
lidar_sensors = flatten(

0 commit comments

Comments
 (0)