diff --git a/jedi/plugins/stdlib.py b/jedi/plugins/stdlib.py index 7321a683..4c723d42 100644 --- a/jedi/plugins/stdlib.py +++ b/jedi/plugins/stdlib.py @@ -819,10 +819,10 @@ class EnumInstance(LazyValueWrapper): def tree_name_to_values(func): - def wrapper(inference_state, value, tree_name): - if tree_name.value == 'sep' and value.is_module() and value.py__name__() == 'os.path': + def wrapper(inference_state, context, tree_name): + if tree_name.value == 'sep' and context.is_module() and context.py__name__() == 'os.path': return ValueSet({ compiled.create_simple_object(inference_state, os.path.sep), }) - return func(inference_state, value, tree_name) + return func(inference_state, context, tree_name) return wrapper