CR changes

Author: mnoszczak

labelbox/schema/model_run.py

@@ -458,29 +458,31 @@ def export_labels(
 
     def export_v2(self,
                   task_name: Optional[str] = None,
-                  params: Optional[ModelRunExportParams] = {}) -> Task:
+                  params: Optional[ModelRunExportParams] = None) -> Task:
         mutation_name = "exportDataRowsInModelRun"
         create_task_query_str = """mutation exportDataRowsInModelRunPyApi($input: ExportDataRowsInModelRunInput!){
           %s(input: $input) {taskId} }
           """ % (mutation_name)
         if (task_name is None):
             task_name = f'Export Data Rows in Model Run - {self.name}'
 
-        params = {
+        _params = params or ModelRunExportParams()
+
+        queryParams = {
             "input": {
                 "taskName": task_name,
                 "filters": {
                     "modelRunId": self.uid
                 },
                 "params": {
                     "includeAttachments":
-                        params.get('attachments', False),
+                        _params.get('attachments', False),
                     "includeMediaAttributes":
-                        params.get('media_attributes', False),
+                        _params.get('media_attributes', False),
                     "includeMetadata":
-                        params.get('metadata_fields', False),
+                        _params.get('metadata_fields', False),
                     "includeDataRowDetails":
-                        params.get('data_row_details', False),
+                        _params.get('data_row_details', False),
                     # Arguments locked based on exectuion context
                     "includeProjectDetails":
                         False,
@@ -493,7 +495,7 @@ def export_v2(self,
         }
         res = self.client.execute(
             create_task_query_str,
-            params,
+            queryParams,
         )
         res = res[mutation_name]
         task_id = res["taskId"]

labelbox/schema/project.py

@@ -375,15 +375,6 @@ def _validate_datetime(string_date: str) -> bool:
                          self.uid)
             time.sleep(sleep_time)
 
-    defaultExportParams: ProjectExportParams = {
-        "attachments": False,
-        "media_attributes": False,
-        "metadata_fields": False,
-        "data_row_details": False,
-        "project_details": False,
-        "labels": False,
-        "performance_details": False
-    }
     """
     Creates a project run export task with the given params and returns the task.
     
@@ -393,11 +384,21 @@ def _validate_datetime(string_date: str) -> bool:
 
     def export_v2(self,
                   task_name: Optional[str] = None,
-                  params: ProjectExportParams = defaultExportParams) -> Task:
+                  params: Optional[ProjectExportParams] = None) -> Task:
 
         if (task_name is None):
             task_name = f'Export Data Rows in Project - {self.name}'
 
+        _params = params or ProjectExportParams({
+            "attachments": False,
+            "media_attributes": False,
+            "metadata_fields": False,
+            "data_row_details": False,
+            "project_details": False,
+            "labels": False,
+            "performance_details": False
+        })
+
         mutation_name = "exportDataRowsInProject"
         create_task_query_str = """mutation exportDataRowsInProjectPyApi($input: ExportDataRowsInProjectInput!){
           %s(input: $input) {taskId} }
@@ -410,19 +411,19 @@ def export_v2(self,
                 },
                 "params": {
                     "includeAttachments":
-                        params.get('attachments', False),
+                        _params.get('attachments', False),
                     "includeMediaAttributes":
-                        params.get('media_attributes', False),
+                        _params.get('media_attributes', False),
                     "includeMetadata":
-                        params.get('metadata_fields', False),
+                        _params.get('metadata_fields', False),
                     "includeDataRowDetails":
-                        params.get('data_row_details', False),
+                        _params.get('data_row_details', False),
                     "includeProjectDetails":
-                        params.get('project_details', False),
+                        _params.get('project_details', False),
                     "includeLabels":
-                        params.get('labels', False),
+                        _params.get('labels', False),
                     "includePerformanceDetails":
-                        params.get('performance_details', False),
+                        _params.get('performance_details', False),
                 },
             }
         }