Support include predictions in catalog exports (#1935)

mnoszczak · web-flow · commit a9d9360c1578 · 2025-01-02T11:05:26.000-08:00
diff --git a/libs/labelbox/src/labelbox/schema/catalog.py b/libs/labelbox/src/labelbox/schema/catalog.py
@@ -104,6 +104,7 @@ def _export(
                 "interpolated_frames": False,
                 "all_projects": False,
                 "all_model_runs": False,
+                "predictions": False,
             }
         )
         validate_catalog_export_params(_params)
diff --git a/libs/labelbox/src/labelbox/schema/data_row.py b/libs/labelbox/src/labelbox/schema/data_row.py
@@ -315,6 +315,7 @@ def _export(
                 "interpolated_frames": False,
                 "all_projects": False,
                 "all_model_runs": False,
+                "predictions": False,
             }
         )
 
diff --git a/libs/labelbox/src/labelbox/schema/dataset.py b/libs/labelbox/src/labelbox/schema/dataset.py
@@ -392,6 +392,7 @@ def _export(
                 "interpolated_frames": False,
                 "all_projects": False,
                 "all_model_runs": False,
+                "predictions": False,
             }
         )
         validate_catalog_export_params(_params)
diff --git a/libs/labelbox/src/labelbox/schema/export_params.py b/libs/labelbox/src/labelbox/schema/export_params.py
@@ -29,6 +29,7 @@ class CatalogExportParams(DataRowParams):
     interpolated_frames: Optional[bool]
     all_projects: Optional[bool]
     all_model_runs: Optional[bool]
+    predictions: Optional[bool]
 
 
 class ModelRunExportParams(DataRowParams):
diff --git a/libs/labelbox/src/labelbox/schema/slice.py b/libs/labelbox/src/labelbox/schema/slice.py
@@ -160,6 +160,7 @@ def _export(
                 "interpolated_frames": False,
                 "all_projects": False,
                 "all_model_runs": False,
+                "predictions": False,
             }
         )
         validate_catalog_export_params(_params)

Original file line number	Diff line number	Diff line change
`@@ -104,6 +104,7 @@ def _export(`
`104`	`104`	`"interpolated_frames": False,`
`105`	`105`	`"all_projects": False,`
`106`	`106`	`"all_model_runs": False,`
	`107`	`+ "predictions": False,`
`107`	`108`	`}`
`108`	`109`	`)`
`109`	`110`	`validate_catalog_export_params(_params)`
Original file line number	Diff line number	Diff line change
`@@ -315,6 +315,7 @@ def _export(`
`315`	`315`	`"interpolated_frames": False,`
`316`	`316`	`"all_projects": False,`
`317`	`317`	`"all_model_runs": False,`
	`318`	`+ "predictions": False,`
`318`	`319`	`}`
`319`	`320`	`)`
`320`	`321`
Original file line number	Diff line number	Diff line change
`@@ -392,6 +392,7 @@ def _export(`
`392`	`392`	`"interpolated_frames": False,`
`393`	`393`	`"all_projects": False,`
`394`	`394`	`"all_model_runs": False,`
	`395`	`+ "predictions": False,`
`395`	`396`	`}`
`396`	`397`	`)`
`397`	`398`	`validate_catalog_export_params(_params)`
Original file line number	Diff line number	Diff line change
`@@ -160,6 +160,7 @@ def _export(`
`160`	`160`	`"interpolated_frames": False,`
`161`	`161`	`"all_projects": False,`
`162`	`162`	`"all_model_runs": False,`
	`163`	`+ "predictions": False,`
`163`	`164`	`}`
`164`	`165`	`)`
`165`	`166`	`validate_catalog_export_params(_params)`