PR feedback: make UploadManifest a pydantic model

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/dataset.py

@@ -620,7 +620,7 @@ def _exec_upsert_data_rows(
             file_upload_thread_count=file_upload_thread_count,
             upsert_chunk_size=UPSERT_CHUNK_SIZE)
 
-        data = json.dumps(manifest.to_dict()).encode("utf-8")
+        data = json.dumps(manifest.dict()).encode("utf-8")
         manifest_uri = self.client.upload_data(data,
                                                content_type="application/json",
                                                filename="manifest.json")

libs/labelbox/src/labelbox/schema/internal/data_row_uploader.py

@@ -10,23 +10,15 @@
 from labelbox.orm.model import Entity
 from labelbox.orm.model import Field
 from labelbox.schema.embedding import EmbeddingVector
+from labelbox.pydantic_compat import BaseModel
 from labelbox.schema.internal.datarow_upload_constants import MAX_DATAROW_PER_API_OPERATION
 from labelbox.schema.internal.data_row_upsert_item import DataRowUpsertItem
 
 
-class UploadManifest:
-
-    def __init__(self, source: str, item_count: int, chunk_uris: List[str]):
-        self.source = source
-        self.item_count = item_count
-        self.chunk_uris = chunk_uris
-
-    def to_dict(self):
-        return {
-            "source": self.source,
-            "item_count": self.item_count,
-            "chunk_uris": self.chunk_uris
-        }
+class UploadManifest(BaseModel):
+    source: str
+    item_count: int
+    chunk_uris: List[str]
 
 
 class DataRowUploader: