Revert back to TypedDict

Author: mnoszczak

labelbox/schema/export_params.py

@@ -1,23 +1,25 @@
 import sys
 
 from typing import Optional
+if sys.version_info >= (3, 8):
+    from typing import TypedDict
+else:
+    from typing_extensions import TypedDict
 
-from pydantic import BaseModel
 
+class DataRowParams(TypedDict):
+    data_row_details: Optional[bool]
+    media_attributes: Optional[bool]
+    metadata_fields: Optional[bool]
+    attachments: Optional[bool]
 
-class DataRowParams(BaseModel):
-    data_row_details: Optional[bool] = None
-    media_attributes: Optional[bool] = None
-    metadata_fields: Optional[bool] = None
-    attachments: Optional[bool] = None
 
+class ProjectExportParams(DataRowParams):
+    project_details: Optional[bool]
+    label_details: Optional[bool]
+    performance_details: Optional[bool]
 
-class ProjectExportParams(BaseModel):
-    include_project_details: Optional[bool] = None
-    include_label_details: Optional[bool] = None
-    include_performance_details: Optional[bool] = None
 
-
-class ModelRunExportParams(BaseModel):
+class ModelRunExportParams(DataRowParams):
     # TODO: Add model run fields
     pass