From 89282ba0c65df8dd83e216d29f811131726998e8 Mon Sep 17 00:00:00 2001 From: "Lewis Sword (lew-sword)" Date: Tue, 10 Jun 2025 11:12:15 +0100 Subject: [PATCH] GEN-1501 bug fix: adding missing max_rows parameter to BuildDataset in SyncGravity --- pyproject.toml | 2 +- src/macrocosmos/resources/gravity.py | 3 +++ uv.lock | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 45eafc9..ec0ae12 100644 --- a/pyproject.toml +++ b/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" diff --git a/src/macrocosmos/resources/gravity.py b/src/macrocosmos/resources/gravity.py index 7d88058..ca71016 100644 --- a/src/macrocosmos/resources/gravity.py +++ b/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, ) ) diff --git a/uv.lock b/uv.lock index d6f9fa5..9bbe441 100644 --- a/uv.lock +++ b/uv.lock @@ -556,7 +556,7 @@ wheels = [ [[package]] name = "macrocosmos" -version = "1.0.4" +version = "1.0.5" source = { editable = "." } dependencies = [ { name = "grpcio", version = "1.70.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },