AILBlockWalkerBase.walk_statement takes block as an input arg. (#280)

angr · Feb 7, 2025 · b2e3f17 · b2e3f17
1 parent cbfa30f
commit b2e3f17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ailment/block_walker.py b/ailment/block_walker.py
@@ -68,8 +68,8 @@ def walk(self, block: Block) -> None:
             self._handle_stmt(i, stmt, block)
             i += 1
 
-    def walk_statement(self, stmt: Statement):
-        return self._handle_stmt(0, stmt, None)
+    def walk_statement(self, stmt: Statement, block: Block | None = None):
+        return self._handle_stmt(0, stmt, block)
 
     def walk_expression(
         self,