refactor for mypy

Author: BeArchiTek

infrahub_sdk/client.py

@@ -566,17 +566,20 @@ async def count(
         **kwargs: Any,
     ) -> int:
         """Return the number of nodes of a given kind."""
-        filters = kwargs
-        schema = await self.schema.get(kind=kind, branch=branch)
+        filters: dict[str, Any] = dict(kwargs)
+
+        if partial_match:
+            filters["partial_match"] = True
 
+        schema = await self.schema.get(kind=kind, branch=branch)
         branch = branch or self.default_branch
         if at:
             at = Timestamp(at)
 
-        data = {"count": None, "@filters": filters}
-
-        if partial_match:
-            data["@filters"]["partial_match"] = True
+        data: dict[str, Any] = {
+            "count": None,
+            "@filters": filters,
+        }
 
         response = await self.execute_graphql(
             query=Query(query={schema.kind: data}).render(),
@@ -1693,16 +1696,20 @@ def count(
         **kwargs: Any,
     ) -> int:
         """Return the number of nodes of a given kind."""
-        filters = kwargs
-        schema = self.schema.get(kind=kind, branch=branch)
+        filters: dict[str, Any] = dict(kwargs)
+
+        if partial_match:
+            filters["partial_match"] = True
 
+        schema = self.schema.get(kind=kind, branch=branch)
         branch = branch or self.default_branch
         if at:
             at = Timestamp(at)
 
-        data = {"count": None, "@filters": filters}
-        if partial_match:
-            data["@filters"]["partial_match"] = True
+        data: dict[str, Any] = {
+            "count": None,
+            "@filters": filters,
+        }
 
         response = self.execute_graphql(
             query=Query(query={schema.kind: data}).render(),
@@ -1712,6 +1719,7 @@ def count(
         )
         return int(response.get(schema.kind, {}).get("count", 0))
 
+
     @overload
     def all(
         self,