diff --git a/pyteal/ast/continue_.py b/pyteal/ast/continue_.py index 1954471c0..a134925bd 100644 --- a/pyteal/ast/continue_.py +++ b/pyteal/ast/continue_.py @@ -28,7 +28,7 @@ def __teal__(self, options: "CompileOptions"): if not options.isInLoop(): raise TealCompileError("continue is only allowed in a loop", self) - start = TealSimpleBlock([], root_expr=self) + start = TealSimpleBlock([]) options.addLoopContinueBlock(start) return start, start diff --git a/pyteal/ir/tealsimpleblock.py b/pyteal/ir/tealsimpleblock.py index 18009da6f..18ef7e217 100644 --- a/pyteal/ir/tealsimpleblock.py +++ b/pyteal/ir/tealsimpleblock.py @@ -5,8 +5,8 @@ class TealSimpleBlock(TealBlock): """Represents a basic block of TealComponents in a graph that does not contain a branch condition.""" - def __init__(self, ops: list[TealOp], root_expr: "Expr | None" = None) -> None: # type: ignore - super().__init__(ops, root_expr=root_expr) + def __init__(self, ops: list[TealOp]) -> None: + super().__init__(ops) self.nextBlock: TealBlock | None = None self.visited = False