move new fields from base object to annotation

Author: jtsodapop

labelbox/data/serialization/ndjson/base.py

@@ -35,6 +35,8 @@ class Config:
 class NDAnnotation(NDJsonBase):
     name: Optional[str] = None
     schema_id: Optional[Cuid] = None
+    page: Optional[int] = None
+    unit: Optional[str] = None
 
     @root_validator()
     def must_set_one(cls, values):
@@ -50,4 +52,4 @@ def dict(self, *args, **kwargs):
             res.pop('name')
         if 'schemaId' in res and res['schemaId'] is None:
             res.pop('schemaId')
-        return res
+        return res

labelbox/data/serialization/ndjson/objects.py

@@ -21,8 +21,6 @@
 
 class NDBaseObject(NDAnnotation):
     classifications: List[NDSubclassificationType] = []
-    page: Optional[int] = None
-    unit: Optional[str] = None
 
 
 class VideoSupported(BaseModel):