From c9bfc395cd3a0cd9a2d7e1e2d1c1a4d69ee369ab Mon Sep 17 00:00:00 2001 From: Lukasz Gluszek Date: Sat, 31 Jan 2026 04:08:29 +0100 Subject: [PATCH] Prelabels for text subclasses under global text classes --- .../classification/classification.py | 1 + .../serialization/ndjson/classification.py | 18 ++++++++++++++++++ 2 files changed, 19 insertions(+) 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, )