override dicts

Author: farkob

labelbox/data/annotation_types/feature.py

@@ -24,3 +24,11 @@ def must_set_one(cls, values):
                 "Must set either feature_schema_id or name for all feature schemas"
             )
         return values
+
+    def dict(self, *args, **kwargs):
+        res = super().dict(*args, **kwargs)
+        if res.name is None:
+            res.pop('name')
+        if res.feature_schema_id is None:
+            res.pop('feature_schema_id')
+        return res

labelbox/data/serialization/ndjson/base.py

@@ -47,4 +47,12 @@ def validate_id(cls, v, values):
             raise ValueError(
                 "Schema ids or names are not set. Use `LabelGenerator.assign_feature_schema_ids`, `LabelList.assign_feature_schema_ids`, or `Label.assign_feature_schema_ids`."
             )
-        return v
+        return v
+
+    def dict(self, *args, **kwargs):
+        res = super().dict(*args, **kwargs)
+        if res.name is None:
+            res.pop('name')
+        if res.schema_id is None:
+            res.pop('schema_id')
+        return res

labelbox/data/serialization/ndjson/classification.py

@@ -27,6 +27,14 @@ def validate_id(cls, v, values):
             )
         return v
 
+    def dict(self, *args, **kwargs):
+        res = super().dict(*args, **kwargs)
+        if res.name is None:
+            res.pop('name')
+        if res.schema_id is None:
+            res.pop('schema_id')
+        return res
+
     class Config:
         allow_population_by_field_name = True
         alias_generator = camel_case

labelbox/data/serialization/ndjson/converter.py

@@ -39,9 +39,4 @@ def serialize(
             A generator for accessing the ndjson representation of the data
         """
         for example in NDLabel.from_common(labels):
-            res = example.dict(by_alias=True)
-            if res.schema_id is None:
-                res.pop('schema_id')
-            if res.name is None:
-                res.pop('name')
-            yield res
+            yield example.dict(by_alias=True)