Fix

execute.py

@@ -28,7 +28,7 @@ import subprocess
 import tempfile
 
 # https://github.com/THUDM/CodeGeeX/blob/ebeb850f227a90c79de39f7e26b1302f374f3240/codegeex/benchmark/rust/Cargo.toml
-BASE_CARGO = """[package]
+BASE_CARGO = '''[package]
 name = "rust"
 version = "0.1.0"
 edition = "2021"
@@ -38,8 +38,8 @@ edition = "2021"
 [dependencies]
 rand = "0.4"
 regex = "1"
-md5 = "0.7.0
-"""
+md5 = "0.7.0"
+'''
 
 
 def check_correctness(check_program, timeout, task_id, completion_id, language):
@@ -273,9 +273,9 @@ def unsafe_execute_rust(check_program, result, timeout):
         # Proceed towards Rust binaries compilation. Therefore move to Rust module root dir.
         os.chdir(RUST_DIR)
 
-        compilation_result = subprocess.run(["cargo", "check", "--bin", "test", "--message-format", "json"], capture_output=True)
+        compilation_result = subprocess.run(["cargo", "check", "--bin", "test", "--message-format", "json"], timeout=timeout, capture_output=True)
         if compilation_result.returncode == 0:
-            exec_result = subprocess.run(["cargo", "test", "--bin", "test", "--message-format", "json"], capture_output=True)
+            exec_result = subprocess.run(["cargo", "test", "--bin", "test", "--message-format", "json"], timeout=timeout, capture_output=True)
             if exec_result.returncode == 0:
                 result.append("passed")
             else: