From d28e04ce1acbedccd37503b249447b8f969ff4aa Mon Sep 17 00:00:00 2001 From: cyc60 Date: Thu, 19 Dec 2024 18:46:39 +0300 Subject: [PATCH] Small graph client updates --- pyproject.toml | 2 +- sw_utils/graph/__init__.py | 2 ++ sw_utils/graph/client.py | 4 ++-- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 3ea1092..0fcb7cb 100644 --- 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 "] license = "GPL-3.0-or-later" diff --git a/sw_utils/graph/__init__.py b/sw_utils/graph/__init__.py index e69de29..71ac729 100644 --- 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 index 3543ed8..18f881d 100644 --- 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. """