Include predictions (dataset export) (#1689)

vbrodsky · web-flow · commit 3680d3831c59 · 2024-06-24T14:28:21.000-04:00
diff --git a/libs/labelbox/src/labelbox/schema/dataset.py b/libs/labelbox/src/labelbox/schema/dataset.py
@@ -545,6 +545,8 @@ def _export(
                         _params.get('label_details', False),
                     "includeInterpolatedFrames":
                         _params.get('interpolated_frames', False),
+                    "includePredictions":
+                        _params.get('predictions', False),
                     "projectIds":
                         _params.get('project_ids', None),
                     "modelRunIds":
