diff options
Diffstat (limited to 'astroid/brain/brain_functools.py')
-rw-r--r-- | astroid/brain/brain_functools.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/astroid/brain/brain_functools.py b/astroid/brain/brain_functools.py index ffdbc888..f6a9830d 100644 --- a/astroid/brain/brain_functools.py +++ b/astroid/brain/brain_functools.py @@ -18,7 +18,7 @@ from astroid.interpreter import objectmodel from astroid.manager import AstroidManager from astroid.nodes.node_classes import AssignName, Attribute, Call, Name from astroid.nodes.scoped_nodes import FunctionDef -from astroid.util import Uninferable +from astroid.util import UninferableBase LRU_CACHE = "functools.lru_cache" @@ -84,7 +84,7 @@ def _functools_partial_inference( inferred_wrapped_function = next(partial_function.infer(context=context)) except (InferenceError, StopIteration) as exc: raise UseInferenceDefault from exc - if inferred_wrapped_function is Uninferable: + if isinstance(inferred_wrapped_function, UninferableBase): raise UseInferenceDefault("Cannot infer the wrapped function") if not isinstance(inferred_wrapped_function, FunctionDef): raise UseInferenceDefault("The wrapped function is not a function") |