diff --git a/labelbox/schema/export_params.py b/labelbox/schema/export_params.py index 6da97b17b..67f6977e9 100644 --- a/labelbox/schema/export_params.py +++ b/labelbox/schema/export_params.py @@ -38,6 +38,7 @@ class CatalogExportParams(DataRowParams): class ModelRunExportParams(DataRowParams): predictions: Optional[bool] + model_run_details: Optional[bool] def _validate_array_length(array, max_length, array_name): diff --git a/labelbox/schema/model_run.py b/labelbox/schema/model_run.py index b2de493c7..f57bbf132 100644 --- a/labelbox/schema/model_run.py +++ b/labelbox/schema/model_run.py @@ -569,6 +569,8 @@ def _export( _params.get('data_row_details', False), "includePredictions": _params.get('predictions', False), + "includeModelRunDetails": + _params.get('model_run_details', False), }, "streamable": streamable }