Fix typing issues

Author: Cito

src/graphql/language/parser.py

@@ -1,6 +1,7 @@
-from typing import Callable, Dict, List, Optional, Union, TypeVar, cast
 from functools import partial
+from typing import Callable, Dict, List, Optional, TypeVar, Union, cast
 
+from ..error import GraphQLError, GraphQLSyntaxError
 from .ast import (
     ArgumentNode,
     BooleanValueNode,
@@ -24,14 +25,14 @@
     InputObjectTypeDefinitionNode,
     InputObjectTypeExtensionNode,
     InputValueDefinitionNode,
-    IntValueNode,
     InterfaceTypeDefinitionNode,
     InterfaceTypeExtensionNode,
+    IntValueNode,
     ListTypeNode,
     ListValueNode,
     Location,
-    NameNode,
     NamedTypeNode,
+    NameNode,
     NonNullTypeNode,
     NullValueNode,
     ObjectFieldNode,
@@ -48,6 +49,7 @@
     SelectionNode,
     SelectionSetNode,
     StringValueNode,
+    Token,
     TypeNode,
     TypeSystemExtensionNode,
     UnionTypeDefinitionNode,
@@ -57,11 +59,9 @@
     VariableNode,
 )
 from .directive_locations import DirectiveLocation
-from .ast import Token
 from .lexer import Lexer, is_punctuator_token_kind
 from .source import Source, is_source
 from .token_kind import TokenKind
-from ..error import GraphQLError, GraphQLSyntaxError
 
 __all__ = ["parse", "parse_type", "parse_value", "parse_const_value"]
 
@@ -401,8 +401,9 @@ def parse_field(self) -> FieldNode:
     def parse_arguments(self, is_const: bool) -> List[ArgumentNode]:
         """Arguments[Const]: (Argument[?Const]+)"""
         item = self.parse_const_argument if is_const else self.parse_argument
-        item = cast(Callable[[], ArgumentNode], item)
-        return self.optional_many(TokenKind.PAREN_L, item, TokenKind.PAREN_R)
+        return self.optional_many(
+            TokenKind.PAREN_L, cast(Callable[[], ArgumentNode], item), TokenKind.PAREN_R
+        )
 
     def parse_argument(self, is_const: bool = False) -> ArgumentNode:
         """Argument[Const]: Name : Value[?Const]"""

src/graphql/language/visitor.py

@@ -14,8 +14,7 @@
 
 from ..pyutils import inspect, snake_to_camel
 from . import ast
-
-from .ast import Node, QUERY_DOCUMENT_KEYS
+from .ast import QUERY_DOCUMENT_KEYS, Node
 
 __all__ = [
     "Visitor",
@@ -288,7 +287,7 @@ def visit(
         else:
             stack = Stack(in_array, idx, keys, edits, stack)
             in_array = isinstance(node, tuple)
-            keys = node if in_array else visitor_keys.get(node.kind, ())
+            keys = node if in_array else visitor_keys.get(node.kind, ())  # type: ignore
             idx = -1
             edits = []
             if parent:

tests/execution/test_abstract.py

@@ -3,7 +3,7 @@
 
 from pytest import mark
 
-from graphql.execution import execute, execute_sync, ExecutionResult
+from graphql.execution import ExecutionResult, execute, execute_sync
 from graphql.language import parse
 from graphql.type import (
     GraphQLBoolean,
@@ -40,9 +40,7 @@ async def execute_query(
     assert isinstance(schema, GraphQLSchema)
     assert isinstance(query, str)
     document = parse(query)
-    result = (execute_sync if sync else execute)(
-        schema, document, root_value
-    )  # type: ignore
+    result = (execute_sync if sync else execute)(schema, document, root_value)
     if not sync and isawaitable(result):
         result = await result
     assert isinstance(result, ExecutionResult)