Upload codeexecutor.py with huggingface_hub

codeexecutor.py

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