Add support for deserialization of confidence for classification types

Author: Val Brodsky

labelbox/data/serialization/ndjson/classification.py

@@ -226,7 +226,8 @@ def to_common(
             value=annotation.to_common(),
             name=annotation.name,
             feature_schema_id=annotation.schema_id,
-            extra={'uuid': annotation.uuid})
+            extra={'uuid': annotation.uuid},
+            confidence=annotation.confidence)
         if getattr(annotation, 'frames', None) is None:
             return [common]
         results = []

tests/data/annotation_types/test_text.py

@@ -0,0 +1,12 @@
+from labelbox.data.annotation_types.classification.classification import Text
+
+
+def test_text():
+    text_entity = Text(answer="good job")
+    assert text_entity.answer == "good job"
+
+
+def test_text_confidence():
+    text_entity = Text(answer="good job", confidence=0.5)
+    assert text_entity.answer == "good job"
+    assert text_entity.confidence == 0.5

tests/data/serialization/ndjson/test_checklist.py

@@ -0,0 +1,43 @@
+import json
+from labelbox.data.annotation_types.annotation import ClassificationAnnotation
+from labelbox.data.annotation_types.classification.classification import Checklist, ClassificationAnswer, Radio
+from labelbox.data.annotation_types.data.text import TextData
+from labelbox.data.annotation_types.label import Label
+
+from labelbox.data.serialization.ndjson.converter import NDJsonConverter
+
+
+def test_serialization():
+    label = Label(uid="ckj7z2q0b0000jx6x0q2q7q0d",
+                  data=TextData(
+                      uid="bkj7z2q0b0000jx6x0q2q7q0d",
+                      text="This is a test",
+                  ),
+                  annotations=[
+                      ClassificationAnnotation(
+                          name="checkbox_question_geo",
+                          confidence=0.5,
+                          value=Checklist(answer=[
+                              ClassificationAnswer(name="first_answer"),
+                              ClassificationAnswer(name="second_answer")
+                          ]))
+                  ])
+
+    serialized = NDJsonConverter.serialize([label])
+
+    res = next(serialized)
+    assert res['confidence'] == 0.5
+    assert res['name'] == "checkbox_question_geo"
+    assert res['answer'][0]['name'] == "first_answer"
+    assert res['answer'][1]['name'] == "second_answer"
+    assert res['dataRow']['id'] == "bkj7z2q0b0000jx6x0q2q7q0d"
+
+    deserialized = NDJsonConverter.deserialize([res])
+    res = next(deserialized)
+    annotation = res.annotations[0]
+    assert annotation.confidence == 0.5
+
+    annotation_value = annotation.value
+    assert type(annotation_value) is Checklist
+    assert annotation_value.answer[0].name == "first_answer"
+    assert annotation_value.answer[1].name == "second_answer"

tests/data/serialization/ndjson/test_radio.py

@@ -0,0 +1,39 @@
+import json
+from labelbox.data.annotation_types.annotation import ClassificationAnnotation
+from labelbox.data.annotation_types.classification.classification import ClassificationAnswer, Radio
+from labelbox.data.annotation_types.data.text import TextData
+from labelbox.data.annotation_types.label import Label
+
+from labelbox.data.serialization.ndjson.converter import NDJsonConverter
+
+
+def test_serialization():
+    label = Label(uid="ckj7z2q0b0000jx6x0q2q7q0d",
+                  data=TextData(
+                      uid="bkj7z2q0b0000jx6x0q2q7q0d",
+                      text="This is a test",
+                  ),
+                  annotations=[
+                      ClassificationAnnotation(
+                          name="radio_question_geo",
+                          confidence=0.5,
+                          value=Radio(answer=ClassificationAnswer(
+                              confidence=0.6, name="first_radio_answer")))
+                  ])
+
+    serialized = NDJsonConverter.serialize([label])
+    res = next(serialized)
+    assert res['confidence'] == 0.5
+    assert res['name'] == "radio_question_geo"
+    assert res['answer']['name'] == "first_radio_answer"
+    assert res['answer']['confidence'] == 0.6
+    assert res['dataRow']['id'] == "bkj7z2q0b0000jx6x0q2q7q0d"
+
+    deserialized = NDJsonConverter.deserialize([res])
+    res = next(deserialized)
+    annotation = res.annotations[0]
+    assert annotation.confidence == 0.5
+
+    annotation_value = annotation.value
+    assert type(annotation_value) is Radio
+    assert annotation_value.answer.name == "first_radio_answer"

tests/data/serialization/ndjson/test_text.py

@@ -26,3 +26,12 @@ def test_serialization():
     assert res['name'] == "radio_question_geo"
     assert res['answer'] == "first_radio_answer"
     assert res['dataRow']['id'] == "bkj7z2q0b0000jx6x0q2q7q0d"
+
+    deserialized = NDJsonConverter.deserialize([res])
+    res = next(deserialized)
+    annotation = res.annotations[0]
+    assert annotation.confidence == 0.5
+
+    annotation_value = annotation.value
+    assert type(annotation_value) is Text
+    assert annotation_value.answer == "first_radio_answer"