Fix mypy issues

Author: Cito

gql/transport/__init__.py

@@ -3,15 +3,15 @@
 
 import six
 from graphql.execution import ExecutionResult
-from graphql.language.ast import Node, Document
+from graphql.language.ast import Document
 from promise import Promise
 
 
 @six.add_metaclass(abc.ABCMeta)
 class Transport:
     @abc.abstractmethod
     def execute(self, document):
-        # type: (Union[Node, Document]) -> Union[ExecutionResult, Promise[ExecutionResult]]
+        # type: (Document) -> Union[ExecutionResult, Promise[ExecutionResult]]
         """Execute the provided document AST for either a remote or local GraphQL Schema.
 
         :param document: GraphQL query as AST Node or Document object.

gql/transport/local_schema.py

@@ -1,10 +1,11 @@
+from typing import Any, Union
+
 from graphql import GraphQLSchema
 from graphql.execution import ExecutionResult, execute
 from graphql.language.ast import Document
 from promise import Promise
 
 from gql.transport import Transport
-from typing import Any, Union
 
 
 class LocalSchemaTransport(Transport):
@@ -21,7 +22,7 @@ def __init__(
         self.schema = schema
 
     def execute(self, document, *args, **kwargs):
-        # type: (Document, Any, Any) -> Union[ExecutionResult, Promise[ExecutionResult]]
+        # type: (Document, *Any, **Any) -> Union[ExecutionResult, Promise[ExecutionResult]]
         """Execute the given document against the configured local schema.
 
         :param document: GraphQL query as AST Node object.

gql/transport/requests.py

@@ -1,14 +1,15 @@
 from __future__ import absolute_import
 
+from typing import Any, Dict, Union
+
 import requests
 from graphql.execution import ExecutionResult
-from graphql.language.ast import Node
+from graphql.language.ast import Document
 from graphql.language.printer import print_ast
 from requests.auth import AuthBase
 from requests.cookies import RequestsCookieJar
 
 from gql.transport import Transport
-from typing import Any, Dict, Union
 
 
 class RequestsHTTPTransport(Transport):
@@ -55,7 +56,7 @@ def __init__(
         self.kwargs = kwargs
 
     def execute(self, document, variable_values=None, timeout=None):
-        # type: (Node, dict, int) -> ExecutionResult
+        # type: (Document, Dict, int) -> ExecutionResult
         """Execute the provided document AST against the configured remote server.
         This uses the requests library to perform a HTTP POST request to the remote server.
 
@@ -81,7 +82,7 @@ def execute(self, document, variable_values=None, timeout=None):
         # Pass kwargs to requests post method
         post_args.update(self.kwargs)
 
-        response = requests.post(self.url, **post_args)
+        response = requests.post(self.url, **post_args)  # type: ignore
         try:
             result = response.json()
             if not isinstance(result, dict):

tests/starwars/schema.py

@@ -42,16 +42,16 @@
         ),
         "name": GraphQLField(GraphQLString, description="The name of the character."),
         "friends": GraphQLField(
-            GraphQLList(characterInterface),
+            GraphQLList(characterInterface),  # type: ignore
             description="The friends of the character, or an empty list if they have none.",
         ),
         "appearsIn": GraphQLField(
             GraphQLList(episodeEnum), description="Which movies they appear in."
         ),
     },
-    resolve_type=lambda character, *_: humanType
+    resolve_type=lambda character, *_: humanType  # type: ignore
     if getHuman(character.id)
-    else droidType,
+    else droidType,  # type: ignore
 )
 
 humanType = GraphQLObjectType(
@@ -136,7 +136,7 @@
                 "episode": GraphQLArgument(
                     description="If omitted, returns the hero of the whole saga. If "
                     "provided, returns the hero of that particular episode.",
-                    type=episodeEnum,
+                    type=episodeEnum,  # type: ignore
                 )
             },
             resolver=lambda root, info, **args: getHero(args.get("episode")),
@@ -180,7 +180,8 @@
             reviewType,
             args={
                 "episode": GraphQLArgument(
-                    description="Episode to create review", type=episodeEnum,
+                    description="Episode to create review",
+                    type=episodeEnum,  # type: ignore
                 ),
                 "review": GraphQLArgument(
                     description="set alive status", type=reviewInputType,

tests/starwars/test_validation.py

@@ -6,7 +6,7 @@
 
 from .schema import StarWarsSchema
 
-introspection = graphql(StarWarsSchema, introspection_query).data
+introspection = graphql(StarWarsSchema, introspection_query).data  # type: ignore
 
 
 @pytest.fixture