Skip to content

Commit 41dfb29

Browse files
Drew KaulDrew Kaul
authored andcommitted
fix issue with type string to enum conversion
1 parent bb4e865 commit 41dfb29

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

nucleus/scene.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,16 @@ class CameraParams:
4343
@dataclass
4444
class SceneDatasetItem:
4545
url: str
46-
type: DatasetItemType
46+
type: str
4747
reference_id: Optional[str] = None
4848
metadata: Optional[dict] = None
4949
camera_params: Optional[CameraParams] = None
5050

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+
5156
@classmethod
5257
def from_json(cls, payload: dict):
5358
return cls(
@@ -176,7 +181,8 @@ def validate(self):
176181
[
177182
source
178183
for source in frame.items.keys()
179-
if frame.items[source].type == DatasetItemType.POINTCLOUD
184+
if frame.items[source].type
185+
== DatasetItemType.POINTCLOUD.value
180186
]
181187
for frame in self.frames_dict.values()
182188
]

0 commit comments

Comments
 (0)