From cdde1416e791cbcdf15a029d86ee82ceac70eb72 Mon Sep 17 00:00:00 2001 From: Fish Date: Thu, 6 Feb 2025 21:43:03 -0700 Subject: [PATCH] AILBlockWalkerBase.walk_statement takes block as an input arg. --- ailment/block_walker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ailment/block_walker.py b/ailment/block_walker.py index 92af66a..411bbf0 100644 --- 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,