GEN-1501 Bug fix: adding missing max_rows parameter to BuildDataset in SyncGravity (version bump v1.0.5)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "macrocosmos"
-version = "1.0.4"
+version = "1.0.5"
 description = "The official Python SDK for Macrocosmos"
 readme = "README.md"
 license = "Apache-2.0"

src/macrocosmos/resources/gravity.py

@@ -352,6 +352,7 @@ def CreateGravityTask(
     def BuildDataset(
         self,
         crawler_id: str,
+        max_rows: int,
         notification_requests: List[
             Union[gravity_p2p.NotificationRequest, Dict]
         ] = None,
@@ -361,6 +362,7 @@ def BuildDataset(
 
         Args:
             crawler_id: The ID of the crawler to build a dataset for.
+            max_rows: The maximum number of rows to include in the dataset.
             notification_requests: The details of the notifications to be sent (optional).
 
         Returns:
@@ -369,6 +371,7 @@ def BuildDataset(
         return asyncio.run(
             self._async_gravity.BuildDataset(
                 crawler_id=crawler_id,
+                max_rows=max_rows,
                 notification_requests=notification_requests,
             )
         )