remove from dict

Author: farkob

labelbox/data/annotation_types/feature.py

@@ -27,8 +27,8 @@ def must_set_one(cls, values):
 
     def dict(self, *args, **kwargs):
         res = super().dict(*args, **kwargs)
-        if self.name is None:
+        if res['name'] is None:
             res.pop('name')
-        if self.feature_schema_id is None:
+        if res['feature_schema_id'] is None:
             res.pop('feature_schema_id')
         return res

labelbox/data/serialization/ndjson/base.py

@@ -51,8 +51,8 @@ def validate_id(cls, v, values):
 
     def dict(self, *args, **kwargs):
         res = super().dict(*args, **kwargs)
-        if self.name is None:
+        if res['name'] is None:
             res.pop('name')
-        if self.schema_id is None:
+        if res['schemaId'] is None:
             res.pop('schemaId')
         return res

labelbox/data/serialization/ndjson/classification.py

@@ -29,9 +29,9 @@ def validate_id(cls, v, values):
 
     def dict(self, *args, **kwargs):
         res = super().dict(*args, **kwargs)
-        if self.name is None:
+        if res['name'] is None:
             res.pop('name')
-        if self.schema_id is None:
+        if res['schemaId'] is None:
             res.pop('schemaId')
         return res