Update codeexecutor.py

codeexecutor.py

@@ -1,4 +1,3 @@
-
 import os
 import re
 import subprocess
@@ -25,8 +24,7 @@ class PythonREPL:
             return True, result.stdout.strip()
         else:
             error_msg = result.stderr.strip()
-            msgs = error_msg.split("
-")
+            msgs = error_msg.split("\n")
             new_msgs = []
             want_next = False
             for m in msgs:
@@ -44,22 +42,16 @@ class PythonREPL:
                 elif want_next:
                     new_msgs.append(m)
                     want_next = False
-            return False, "
-".join(new_msgs).strip()
+            return False, "\n".join(new_msgs).strip()
 
     def __call__(self, query):
-        query = "import math
-import numpy as np
-import sympy as sp
-" + query
-        query = query.strip().split("
-")
+        query = "import math\nimport numpy as np\nimport sympy as sp\n" + query
+        query = query.strip().split("\n")
         if "print(" not in query[-1]:
             if "#" in query[-1]:
                 query[-1] = query[-1].split("#")[0]
             query[-1] = "print(" + query[-1] + ")"
-        query = "
-".join(query)
+        query = "\n".join(query)
         
         with tempfile.TemporaryDirectory() as temp_dir:
             temp_file_path = os.path.join(temp_dir, "tmp.py")
@@ -120,5 +112,4 @@ def get_majority_vote(answers):
         return 0
     c = Counter(answers)
     value, _ = c.most_common()[0]
-    return value
-    
+    return value