Revert back to TypedDict

Author: mnoszczak

labelbox/schema/export_params.py

@@ -1,3 +1,5 @@
+import sys
+
 from typing import Optional
 if sys.version_info >= (3, 8):
     from typing import TypedDict
@@ -14,7 +16,7 @@ class DataRowParams(TypedDict):
 
 class ProjectExportParams(DataRowParams):
     project_details: Optional[bool]
-    label_details: Optional[bool]
+    labels: Optional[bool]
     performance_details: Optional[bool]
 
 

labelbox/schema/project.py

@@ -384,12 +384,21 @@ def _validate_datetime(string_date: str) -> bool:
 
     def export_v2(self, task_name: str,
                   params: Optional[ProjectExportParams]) -> Task:
-        _params = params or {}
+        defaultParams: ProjectExportParams = {
+            "attachments": False,
+            "media_attributes": False,
+            "metadata_fields": False,
+            "data_row_details": False,
+            "project_details": False,
+            "labels": False,
+            "performance_details": False
+        }
+        _params: ProjectExportParams = params if params is not None else defaultParams
         mutation_name = "exportDataRowsInProject"
         create_task_query_str = """mutation exportDataRowsInProjectPyApi($input: ExportDataRowsInProjectInput!){
           %s(input: $input) {taskId} }
           """ % (mutation_name)
-        params = {
+        query_params = {
             "input": {
                 "taskName": task_name,
                 "filters": {
@@ -415,7 +424,7 @@ def export_v2(self, task_name: str,
         }
         res = self.client.execute(
             create_task_query_str,
-            params,
+            query_params,
         )
         res = res[mutation_name]
         task_id = res["taskId"]