CLN: Fix mypy error in pandas/tests/computation/test_eval.py (pandas-…

…dev#29007)
proost · Dec 19, 2019 · 1c081aa · 1c081aa
1 parent ac9ec95
commit 1c081aa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/pandas/tests/computation/test_eval.py b/pandas/tests/computation/test_eval.py
@@ -2,6 +2,7 @@
 from functools import reduce
 from itertools import product
 import operator
+from typing import Dict, Type
 import warnings
 
 import numpy as np
@@ -19,7 +20,11 @@
 from pandas.core.computation.check import _NUMEXPR_VERSION
 from pandas.core.computation.engines import NumExprClobberingError, _engines
 import pandas.core.computation.expr as expr
-from pandas.core.computation.expr import PandasExprVisitor, PythonExprVisitor
+from pandas.core.computation.expr import (
+    BaseExprVisitor,
+    PandasExprVisitor,
+    PythonExprVisitor,
+)
 from pandas.core.computation.expressions import _NUMEXPR_INSTALLED, _USE_NUMEXPR
 from pandas.core.computation.ops import (
     _arith_ops_syms,
@@ -1884,7 +1889,7 @@ def test_invalid_parser():
     "python": PythonExprVisitor,
     "pytables": pytables.ExprVisitor,
     "pandas": PandasExprVisitor,
-}
+}  # type: Dict[str, Type[BaseExprVisitor]]
 
 
 @pytest.mark.parametrize("engine", _engines)

diff --git a/setup.cfg b/setup.cfg
@@ -145,9 +145,6 @@ ignore_errors=True
 [mypy-pandas.tests.arrays.test_period]
 ignore_errors=True
 
-[mypy-pandas.tests.computation.test_eval]
-ignore_errors=True
-
 [mypy-pandas.tests.dtypes.test_common]
 ignore_errors=True