[AL-5142] Add support for predictions param (#1056)

Author: mnoszczak

labelbox/schema/export_params.py

@@ -32,5 +32,4 @@ class CatalogExportParams(DataRowParams):
 
 
 class ModelRunExportParams(DataRowParams):
-    # TODO: Add model run fields
-    pass
+    predictions: Optional[bool]

labelbox/schema/model_run.py

@@ -531,6 +531,8 @@ def export_v2(self,
                         _params.get('metadata_fields', False),
                     "includeDataRowDetails":
                         _params.get('data_row_details', False),
+                    "includePredictions":
+                        _params.get('predictions', False),
                 },
             }
         }

tests/integration/annotation_import/test_model_run.py

@@ -185,7 +185,7 @@ def get_model_run_status():
 def test_model_run_export_v2(model_run_with_data_rows, configured_project):
     task_name = "test_task"
     media_attributes = True
-    params = {"media_attributes": media_attributes}
+    params = {"media_attributes": media_attributes, "predictions": True}
     task_results = _model_run_export_v2_results(model_run_with_data_rows,
                                                 task_name, params)
     label_ids = [label.uid for label in configured_project.labels()]