Handle uuid for ndjson type

Author: Val Brodsky

libs/labelbox/src/labelbox/data/serialization/ndjson/base.py

@@ -20,13 +20,15 @@ def must_set_one(self):
 
 
 class NDJsonBase(_CamelCaseMixin):
-    uuid: str = None
+    uuid: str
     data_row: DataRow
 
-    @field_validator('uuid', mode='before')
+    @model_validator(mode='before')
     @classmethod
-    def set_id(cls, v):
-        return v or str(uuid4())
+    def set_uuid(cls, data):
+        if data.get('uuid') is None:
+            data['uuid'] = str(uuid4())
+        return data
 
     def dict(self, *args, **kwargs):
         """ Pop missing id or missing globalKey from dataRow """

libs/labelbox/src/labelbox/data/serialization/ndjson/converter.py

@@ -113,6 +113,7 @@ def serialize(
                     exclude={"uuid"} if annotation_uuid == "None" else None,
                     exclude_none=True,
                 )
+
                 for k, v in list(res.items()):
                     if k in IGNORE_IF_NONE and v is None:
                         del res[k]

libs/labelbox/tests/data/serialization/ndjson/test_video.py

@@ -23,8 +23,6 @@ def test_video():
     # assert res == [data[2], data[0], data[1], data[3], data[4], data[5]]
     assert (res[0]) == data[2]
     assert (res[1]) == data[0]
-    answers = data[1].pop('answer')
-    data[1]['answers'] = answers
     assert (res[2]) == data[1]
     assert (res[3]) == data[3]
     assert (res[4]) == data[4]