from __future__ import annotations from syntax import EXPRESSION class Optimization: def setup(self, ast: EXPRESSION) -> None: ... def apply(self, node: EXPRESSION) -> EXPRESSION: ...