File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -43,11 +43,16 @@ class CameraParams:
43
43
@dataclass
44
44
class SceneDatasetItem :
45
45
url : str
46
- type : DatasetItemType
46
+ type : str
47
47
reference_id : Optional [str ] = None
48
48
metadata : Optional [dict ] = None
49
49
camera_params : Optional [CameraParams ] = None
50
50
51
+ def __post_init__ (self ):
52
+ assert self .type in [
53
+ e .value for e in DatasetItemType
54
+ ], "type must be one of DatasetItemType's enum values i.e. `image` or `pointcloud`"
55
+
51
56
@classmethod
52
57
def from_json (cls , payload : dict ):
53
58
return cls (
@@ -176,7 +181,8 @@ def validate(self):
176
181
[
177
182
source
178
183
for source in frame .items .keys ()
179
- if frame .items [source ].type == DatasetItemType .POINTCLOUD
184
+ if frame .items [source ].type
185
+ == DatasetItemType .POINTCLOUD .value
180
186
]
181
187
for frame in self .frames_dict .values ()
182
188
]
You can’t perform that action at this time.
0 commit comments