Skip to content

Commit 576c7d5

Browse files
committed
Update coc_plugin.py
fix bug
1 parent ac66567 commit 576c7d5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

optillm/plugins/coc_plugin.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,14 +104,17 @@ def sanitize_code(code: str) -> str:
104104
safe_lines.append(line)
105105

106106
safe_code = '\n'.join(safe_lines)
107+
safe_code = safe_code.replace('\n', '\n ')
107108

108109
# Add safety wrapper
109110
wrapper = f"""
110111
{imports}
112+
111113
def safe_execute():
112114
import numpy as np # Always allow numpy
113-
{safe_code.replace('\\n', '\\n ')}
115+
{safe_code}
114116
return answer if 'answer' in locals() else None
117+
115118
result = safe_execute()
116119
answer = result
117120
"""

0 commit comments

Comments
 (0)