From cc6b8f119223220864ab2988066f64c302273d67 Mon Sep 17 00:00:00 2001 From: cassanof Date: Mon, 28 Aug 2023 16:25:21 -0700 Subject: [PATCH] rstrip codellama --- programming_runs/generators/parse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/programming_runs/generators/parse.py b/programming_runs/generators/parse.py index 08b8da6..38334f0 100644 --- a/programming_runs/generators/parse.py +++ b/programming_runs/generators/parse.py @@ -42,7 +42,7 @@ def parse_first_func(code: str, lang: str) -> Optional[str]: if def_i == -1: return None - return "\n".join(code_lines[def_i:last_i+1]) + return "\n".join(code_lines[def_i:last_i+1]).rstrip("[/PYTHON]") def add_code_block(string: str, lang: str) -> str: