Skip to content

Commit e771541

Browse files
author
Matt Sokoloff
committed
fix error
1 parent d68828e commit e771541

File tree

2 files changed

+3
-33
lines changed

2 files changed

+3
-33
lines changed

labelbox/data/serialization/labelbox_v1/classification.py

Lines changed: 2 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,7 @@ def to_common(self) -> Radio:
2828
@classmethod
2929
def from_common(cls, radio: Radio, feature_schema_id: Cuid,
3030
**extra) -> "LBV1Radio":
31-
return cls(
32-
schema_id=feature_schema_id,
33-
title = radio,
31+
return cls(schema_id=feature_schema_id,
3432
answer=LBV1ClassificationAnswer(
3533
schema_id=radio.answer.feature_schema_id,
3634
title=radio.answer.name,
@@ -62,35 +60,7 @@ def from_common(cls, checklist: Checklist, feature_schema_id: Cuid,
6260
title=answer.name,
6361
value=answer.extra.get('value'),
6462
feature_id=answer.extra.get('feature_id'))
65-
for answer in checklist.answers
66-
],
67-
**extra)
68-
69-
70-
class LBV1Dropdown(LBV1Feature):
71-
answer: List[LBV1ClassificationAnswer]
72-
def to_common(self) -> Dropdown:
73-
return Dropdown(
74-
answer=[
75-
ClassificationAnswer(feature_schema_id=answer.schema_id,
76-
name=answer.title,
77-
extra={
78-
'feature_id': answer.feature_id,
79-
'value': answer.value
80-
}) for answer in self.answer
81-
])
82-
83-
@classmethod
84-
def from_common(cls, dropdown: Dropdown, feature_schema_id: Cuid,
85-
**extra) -> "LBV1Dropdown":
86-
return cls(schema_id = feature_schema_id,
87-
answers=[
88-
LBV1ClassificationAnswer(
89-
schema_id=answer.feature_schema_id,
90-
title=answer.name,
91-
value=answer.extra.get('value'),
92-
feature_id=answer.extra.get('feature_id'))
93-
for answer in dropdown.answer
63+
for answer in checklist.answer
9464
],
9565
**extra)
9666

labelbox/data/serialization/ndjson/classification.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def from_common(
120120
ImageData]) -> "NDChecklist":
121121
return cls(answer=[
122122
NDFeature(schema_id=answer.feature_schema_id)
123-
for answer in checklist.answers
123+
for answer in checklist.answer
124124
],
125125
data_row={'id': data.uid},
126126
schema_id=feature_schema_id,

0 commit comments

Comments
 (0)