diff --git a/jedi/inference/value/dynamic_arrays.py b/jedi/inference/value/dynamic_arrays.py index d94546aa..f888cd36 100644 --- a/jedi/inference/value/dynamic_arrays.py +++ b/jedi/inference/value/dynamic_arrays.py @@ -188,14 +188,14 @@ class _Modification(ValueWrapper): class DictModification(_Modification): - def py__iter__(self): - for lazy_context in self._wrapped_value.py__iter__(): + def py__iter__(self, contextualized_node=None): + for lazy_context in self._wrapped_value.py__iter__(contextualized_node): yield lazy_context yield self._contextualized_key class ListModification(_Modification): - def py__iter__(self): - for lazy_context in self._wrapped_value.py__iter__(): + def py__iter__(self, contextualized_node=None): + for lazy_context in self._wrapped_value.py__iter__(contextualized_node): yield lazy_context yield LazyKnownValues(self._assigned_values)