diff --git a/libs/labelbox/src/labelbox/data/annotation_types/classification/classification.py b/libs/labelbox/src/labelbox/data/annotation_types/classification/classification.py index d6a6448dd..a749c212c 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/classification/classification.py +++ b/libs/labelbox/src/labelbox/data/annotation_types/classification/classification.py @@ -52,6 +52,7 @@ class Text(ConfidenceMixin, CustomMetricsMixin, BaseModel): """ answer: str + classifications: Optional[List["ClassificationAnnotation"]] = None class ClassificationAnnotation( diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/classification.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/classification.py index fedf4d91b..49177bba2 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/classification.py +++ b/libs/labelbox/src/labelbox/data/serialization/ndjson/classification.py @@ -86,6 +86,12 @@ def to_common(self) -> Text: answer=self.answer, confidence=self.confidence, custom_metrics=self.custom_metrics, + classifications=[ + NDSubclassification.to_common(annot) + for annot in self.classifications + ] + if self.classifications + else None, ) @classmethod @@ -98,6 +104,12 @@ def from_common( schema_id=feature_schema_id, confidence=text.confidence, custom_metrics=text.custom_metrics, + classifications=[ + NDSubclassification.from_common(annot) + for annot in text.classifications + ] + if text.classifications + else None, ) @@ -245,6 +257,12 @@ def from_common( message_id=message_id, confidence=text.confidence, custom_metrics=text.custom_metrics, + classifications=[ + NDSubclassification.from_common(annot) + for annot in text.classifications + ] + if text.classifications + else None, )