Use TreeContext in a good way

2018-08-02 11:32:13 +02:00
parent 9a4a96b453
commit 9ff5050d01
9 changed files with 18 additions and 23 deletions
--- a/jedi/evaluate/context/module.py
+++ b/jedi/evaluate/context/module.py
@@ -43,8 +43,11 @@ class ModuleContext(TreeContext):
    parent_context = None

    def __init__(self, evaluator, module_node, path, code_lines):
-        super(ModuleContext, self).__init__(evaluator, parent_context=None)
-        self.tree_node = module_node
+        super(ModuleContext, self).__init__(
+            evaluator,
+            parent_context=None,
+            tree_node=module_node
+        )
        self._path = path
        self.code_lines = code_lines