Update dependencies and fix tests

aorwall · Jul 29, 2024 · 5497215 · 5497215
1 parent aaecd8a
commit 5497215
Show file tree

Hide file tree

Showing 6 changed files with 653 additions and 614 deletions.
diff --git a/.gitignore b/.gitignore
@@ -156,3 +156,5 @@ cython_debug/
 .idea
 notebooks/.ipynb_checkpoints/
 notebooks/local_experiments.ipynb
+
+playground
diff --git a/moatless/codeblocks/codeblocks.py b/moatless/codeblocks/codeblocks.py
@@ -893,10 +893,10 @@ def full_path(self):
     @property
     def module(self) -> "Module":  # noqa: F821
         if self.parent:
-            return self.parent.root()
+            return self.parent.module
         return self
 
-    @deprecated("Use module()")
+    @deprecated("Use codeblock.module")
     def root(self) -> "Module":  # noqa: F821
         return self.module
 

diff --git a/moatless/codeblocks/parser/python.py b/moatless/codeblocks/parser/python.py
@@ -92,7 +92,7 @@ def post_process(self, codeblock: CodeBlock):
                         if rel.type == RelationshipType.IS_A
                     ]
                     if is_a_rel:
-                        super_class = codeblock.root().find_by_path(is_a_rel[0].path)
+                        super_class = codeblock.module.find_by_path(is_a_rel[0].path)
 
                         if super_class:
                             reference.path = (
@@ -132,4 +132,4 @@ def post_process(self, codeblock: CodeBlock):
     def is_outcommented_code(self, comment):
         return comment.startswith("# ...") or any(
             keyword in comment.lower() for keyword in commented_out_keywords
-        )
+        )