Skip to content

Commit 7371bb6

Browse files
Drew KaulDrew Kaul
authored andcommitted
add Scene and LidarScene to init
1 parent 729cd7f commit 7371bb6

File tree

2 files changed

+13
-11
lines changed

2 files changed

+13
-11
lines changed

nucleus/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,7 @@
135135
)
136136
from .slice import Slice
137137
from .upload_response import UploadResponse
138+
from .scene import Scene, LidarScene
138139

139140
# pylint: disable=E1101
140141
# TODO: refactor to reduce this file to under 1000 lines.

nucleus/scene.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -108,17 +108,6 @@ def __post_init__(self):
108108
else:
109109
self.frames_dict = dict(enumerate(self.frames))
110110

111-
# TODO: move validation to scene upload
112-
assert isinstance(self.frames, List), "frames must be a list"
113-
for frame in self.frames:
114-
assert isinstance(
115-
frame, Frame
116-
), "each element of frames must be a Frame object"
117-
assert len(self.frames) > 0, "frames must have length of at least 1"
118-
assert isinstance(
119-
self.reference_id, str
120-
), "reference_id must be a string"
121-
122111
def check_valid_frame_indices(self):
123112
infer_from_list_position = all(
124113
(frame.index is None for frame in self.frames)
@@ -129,6 +118,18 @@ def check_valid_frame_indices(self):
129118
assert (
130119
infer_from_list_position or explicit_frame_order
131120
), "Must specify index explicitly for all frames or implicitly for all frames (inferred from list position)"
121+
122+
# TODO: move validation to scene upload
123+
def validate_scene(self):
124+
assert isinstance(self.frames, List), "frames must be a list"
125+
assert len(self.frames) > 0, "frames must have length of at least 1"
126+
for frame in self.frames:
127+
assert isinstance(
128+
frame, Frame
129+
), "each element of frames must be a Frame object"
130+
assert isinstance(
131+
self.reference_id, str
132+
), "reference_id must be a string"
132133

133134
def add_item(self, item: SceneDatasetItem, index: int, sensor_name: str):
134135
if index not in self.frames_dict:

0 commit comments

Comments
 (0)