From eca008f40d73d9d7afa6f106671afdd87cc0a35c Mon Sep 17 00:00:00 2001 From: Simon Hawkins Date: Tue, 6 Oct 2020 18:17:17 +0100 Subject: [PATCH] TYP: check_untyped_defs core.computation.pytables --- pandas/core/computation/pytables.py | 20 ++++++++++++-------- setup.cfg | 3 --- 2 files changed, 12 insertions(+), 11 deletions(-) diff --git a/pandas/core/computation/pytables.py b/pandas/core/computation/pytables.py index d876c655421ef..dd622ed724e8f 100644 --- a/pandas/core/computation/pytables.py +++ b/pandas/core/computation/pytables.py @@ -42,7 +42,10 @@ class Term(ops.Term): env: PyTablesScope def __new__(cls, name, env, side=None, encoding=None): - klass = Constant if not isinstance(name, str) else cls + if isinstance(name, str): + klass = cls + else: + klass = Constant return object.__new__(klass) def __init__(self, name, env: PyTablesScope, side=None, encoding=None): @@ -83,6 +86,7 @@ class BinOp(ops.BinOp): op: str queryables: Dict[str, Any] + condition: Optional[str] def __init__(self, op: str, lhs, rhs, queryables: Dict[str, Any], encoding): super().__init__(op, lhs, rhs) @@ -184,10 +188,8 @@ def convert_value(self, v) -> "TermValue": def stringify(value): if self.encoding is not None: - encoder = partial(pprint_thing_encoded, encoding=self.encoding) - else: - encoder = pprint_thing - return encoder(value) + return pprint_thing_encoded(value, encoding=self.encoding) + return pprint_thing(value) kind = ensure_decoded(self.kind) meta = ensure_decoded(self.meta) @@ -257,9 +259,11 @@ def __repr__(self) -> str: def invert(self): """ invert the filter """ if self.filter is not None: - f = list(self.filter) - f[1] = self.generate_filter_op(invert=True) - self.filter = tuple(f) + self.filter = ( + self.filter[0], + self.generate_filter_op(invert=True), + self.filter[2], + ) return self def format(self): diff --git a/setup.cfg b/setup.cfg index e125eea226b10..1ddfd900fd1b7 100644 --- a/setup.cfg +++ b/setup.cfg @@ -160,9 +160,6 @@ check_untyped_defs=False [mypy-pandas.core.computation.ops] check_untyped_defs=False -[mypy-pandas.core.computation.pytables] -check_untyped_defs=False - [mypy-pandas.core.computation.scope] check_untyped_defs=False