From 3229cdab4faf9a6d6266085e17e6bbf7565eb73b Mon Sep 17 00:00:00 2001 From: Xinlong Hu Date: Mon, 4 Aug 2025 23:50:59 -0500 Subject: [PATCH] Sanitize lines --- dev_scripts/cflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dev_scripts/cflow.py b/dev_scripts/cflow.py index 7f9ddc8..c702e15 100644 --- a/dev_scripts/cflow.py +++ b/dev_scripts/cflow.py @@ -18,7 +18,7 @@ from rich.progress import Progress, BarColumn, TextColumn, TimeRemainingColumn, from pylingual.control_flow_reconstruction.cfg import CFG from pylingual.control_flow_reconstruction.structure import bc_to_cft from pylingual.main import print_result -from pylingual.control_flow_reconstruction.source import SourceContext +from pylingual.control_flow_reconstruction.source import SourceContext, sanitize_lines from pylingual.editable_bytecode import PYCFile from pylingual.equivalence_check import TestResult, compare_pyc from pylingual.utils.version import PythonVersion @@ -64,7 +64,7 @@ def run(file: Path, out_dir: Path, version: PythonVersion, print=False): file = next(file.iterdir()) in_src = normalize_source(file.read_text(), replace_docstrings=True) - src_lines = in_src.split("\n") + src_lines = sanitize_lines(in_src.split("\n")) in_path = out_dir / "a.py" in_path.write_text(in_src, encoding="utf-8") in_pyc = out_dir / "a.pyc"