Small graph client updates

stakewise · Dec 19, 2024 · d28e04c · d28e04c
1 parent 5a84f89
commit d28e04c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sw-utils"
-version = "v0.7.2"
+version = "v0.7.3"
 description = "StakeWise Python utils"
 authors = ["StakeWise Labs <[email protected]>"]
 license = "GPL-3.0-or-later"

diff --git a/sw_utils/graph/__init__.py b/sw_utils/graph/__init__.py
@@ -0,0 +1,2 @@
+from .client import GraphClient
+from .subgraph_committee import SubgraphCommittee, SubgraphCommitteeVotes
diff --git a/sw_utils/graph/client.py b/sw_utils/graph/client.py
@@ -29,9 +29,9 @@ async def run_query(self, query: DocumentNode, params: dict | None = None) -> di
     async def fetch_pages(
         self,
         query: DocumentNode,
-        page_size: int | None = None,
         params: dict | None = None,
-    ) -> list[dict] | None:
+        page_size: int | None = None,
+    ) -> list[dict]:
         """
         Fetches all pages of the query. Returns concatenated result.
         """
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from .client import GraphClient
		from .subgraph_committee import SubgraphCommittee, SubgraphCommitteeVotes