From 5b9e6d4a5a05df2466b9539b03986a5b90d3e9da Mon Sep 17 00:00:00 2001
From: mnoszczak <mnoszczak@labelbox.com>
Date: Tue, 17 Oct 2023 14:44:58 +0200
Subject: [PATCH] Fix dataset types

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

diff --git a/labelbox/schema/dataset.py b/labelbox/schema/dataset.py
index 9bc63a1f6..1afc482b3 100644
--- a/labelbox/schema/dataset.py
+++ b/labelbox/schema/dataset.py
@@ -68,8 +68,8 @@ class Dataset(DbObject, Updateable, Deletable):
 
     def data_rows(
         self,
-        from_cursor: str = None,
-        where: Comparison = None,
+        from_cursor: Optional[str] = None,
+        where: Optional[Comparison] = None,
     ) -> PaginatedCollection:
         """ 
         Custom method to paginate data_rows via cursor.