From a26ab04b1399f2059844be84c16c559d09bf35d8 Mon Sep 17 00:00:00 2001 From: Val Brodsky Date: Fri, 9 Feb 2024 14:53:43 -0800 Subject: [PATCH] Add model_run_details to export params to support splits etc. --- labelbox/schema/export_params.py | 1 + labelbox/schema/model_run.py | 2 ++ 2 files changed, 3 insertions(+) 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 }