From 3a4eba7ec2e5d1e6034c195d078a32197d9dfbf1 Mon Sep 17 00:00:00 2001 From: Simon Hawkins Date: Sun, 4 Oct 2020 10:33:13 +0100 Subject: [PATCH] TYP: check_untyped_defs compat.pickle_compat pandas\compat\pickle_compat.py:277: error: Incompatible types in assignment (expression has type "Callable[[bytes, DefaultNamedArg(bool, 'fix_imports'), DefaultNamedArg(str, 'encoding'), DefaultNamedArg(str, 'errors')], Any]", variable has type "Callable[[bytes, DefaultNamedArg(bool, 'fix_imports'), DefaultNamedArg(str, 'encoding'), DefaultNamedArg(str, 'errors'), DefaultNamedArg(Optional[Iterable[Any]], 'buffers')], Any]") [assignment] --- pandas/compat/pickle_compat.py | 4 ++-- setup.cfg | 3 --- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/pandas/compat/pickle_compat.py b/pandas/compat/pickle_compat.py index ef9f36705a7ee..80ee1f2e20154 100644 --- a/pandas/compat/pickle_compat.py +++ b/pandas/compat/pickle_compat.py @@ -274,7 +274,7 @@ def patch_pickle(): """ orig_loads = pkl.loads try: - pkl.loads = loads + setattr(pkl, "loads", loads) yield finally: - pkl.loads = orig_loads + setattr(pkl, "loads", orig_loads) diff --git a/setup.cfg b/setup.cfg index 73986f692b6cd..ccabbd830240e 100644 --- a/setup.cfg +++ b/setup.cfg @@ -135,9 +135,6 @@ check_untyped_defs=False [mypy-pandas._version] check_untyped_defs=False -[mypy-pandas.compat.pickle_compat] -check_untyped_defs=False - [mypy-pandas.core.apply] check_untyped_defs=False