Handle default values properly

Author: mnoszczak

labelbox/schema/model_run.py

@@ -456,13 +456,16 @@ def export_labels(
     
     """
 
-    def export_v2(self, task_name: str,
-                  params: Optional[ModelRunExportParams]) -> Task:
-        _params = params or {}
+    def export_v2(self,
+                  task_name: Optional[str] = None,
+                  params: Optional[ModelRunExportParams] = {}) -> 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 = {
             "input": {
                 "taskName": task_name,
@@ -471,13 +474,13 @@ def export_v2(self, task_name: str,
                 },
                 "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,

labelbox/schema/project.py

@@ -375,25 +375,29 @@ 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.
     
     >>>    export_task = export_v2("my_export_task", filter={"media_attributes": True})
     
     """
 
-    def export_v2(self, task_name: str,
-                  params: Optional[ProjectExportParams]) -> Task:
-        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
+    def export_v2(self,
+                  task_name: Optional[str] = None,
+                  params: ProjectExportParams = defaultExportParams) -> Task:
+
+        if (task_name is None):
+            task_name = f'Export Data Rows in Project - {self.name}'
+
         mutation_name = "exportDataRowsInProject"
         create_task_query_str = """mutation exportDataRowsInProjectPyApi($input: ExportDataRowsInProjectInput!){
           %s(input: $input) {taskId} }
@@ -406,19 +410,19 @@ def export_v2(self, task_name: str,
                 },
                 "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),
                 },
             }
         }