rstrip codellama

9 months ago · cc6b8f1192
parent f748264b35
commit cc6b8f1192
1 changed files with 1 additions and 1 deletions
--- 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: