From 1678b576163f17b781b03e9ae0df66b59ded2b49 Mon Sep 17 00:00:00 2001
From: mnoszczak <99751601+mnoszczak@users.noreply.github.com>
Date: Mon, 24 Jun 2024 12:42:49 +0400
Subject: [PATCH] Include predictions (dataset)

---
 libs/labelbox/src/labelbox/schema/dataset.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/libs/labelbox/src/labelbox/schema/dataset.py b/libs/labelbox/src/labelbox/schema/dataset.py
index 081bc949a..c2617cc0d 100644
--- 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":