Directly importing FunctionContext.

jedi/evaluate/base_context.py

@@ -242,7 +242,7 @@ class ContextSet(BaseContextSet):
         return ContextSet.from_iterable(c.py__class__() for c in self._set)
 
     def iterate(self, contextualized_node=None):
-        from jedi.evaluate.context.lazy import get_merged_lazy_context
+        from jedi.evaluate.lazy_context import get_merged_lazy_context
         type_iters = [c.iterate(contextualized_node) for c in self._set]
         for lazy_contexts in zip_longest(*type_iters):
             yield get_merged_lazy_context(