check_untyped_defs pandas.core.computation.expr (part)

Author: simonjayhawkins

pandas/core/computation/expr.py

@@ -7,7 +7,7 @@
 import itertools as it
 import operator
 import tokenize
-from typing import Type
+from typing import Callable, Set, Tuple, Type, TypeVar
 
 import numpy as np
 
@@ -302,7 +302,7 @@ def f(self, *args, **kwargs):
     return f
 
 
-def disallow(nodes):
+def disallow(nodes: Set) -> Callable:
     """Decorator to disallow certain nodes from parsing. Raises a
     NotImplementedError instead.
 
@@ -311,7 +311,9 @@ def disallow(nodes):
     disallowed : callable
     """
 
-    def disallowed(cls):
+    _T = TypeVar("_T", bound="BaseExprVisitor")
+
+    def disallowed(cls: Type[_T]) -> Type[_T]:
         cls.unsupported_nodes = ()
         for node in nodes:
             new_method = _node_not_implemented(node, cls)
@@ -417,6 +419,8 @@ class BaseExprVisitor(ast.NodeVisitor):
         ast.NotIn: ast.NotIn,
     }
 
+    unsupported_nodes: Tuple[str, ...]
+
     def __init__(self, env, engine, parser, preparser=_preparse):
         self.env = env
         self.engine = engine
@@ -712,7 +716,7 @@ def visit_Call(self, node, side=None, **kwargs):
                     "arguments".format(name=res.name)
                 )
 
-            return res(*new_args, **kwargs)
+            return res(*new_args)
 
         else: