Skip to content

Commit 8cf3b04

Browse files
author
JWittmeyer
committed
Change record ide timeout
1 parent e0bcf33 commit 8cf3b04

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

util/record_ide.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def run_record_ide(
4545
knowledge_base_bytes_path = pack_knowledge_base(project_id)
4646

4747
command = [code, record_bytes_path, knowledge_base_bytes_path]
48-
cpu_limit = docker.types.Ulimit(name="cpu", soft=25, hard=25)
48+
cpu_limit = docker.types.Ulimit(name="cpu", soft=50, hard=50)
4949
container_name = str(uuid.uuid4())
5050
container = client.containers.create(
5151
command=command,
@@ -61,10 +61,14 @@ def run_record_ide(
6161
knowledge_base_tar_path = f"{project_id}.tar"
6262

6363
copy_to(
64-
f"./{record_bytes_path}", f"{container.name}:/{record_bytes_path}", record_tar_path
64+
f"./{record_bytes_path}",
65+
f"{container.name}:/{record_bytes_path}",
66+
record_tar_path,
6567
)
6668
copy_to(
67-
f"./{knowledge_base_bytes_path}", f"{container.name}:/{knowledge_base_bytes_path}", knowledge_base_tar_path
69+
f"./{knowledge_base_bytes_path}",
70+
f"{container.name}:/{knowledge_base_bytes_path}",
71+
knowledge_base_tar_path,
6872
)
6973
daemon.run(cancel_container, container_name, container)
7074
__containers_running[container_name] = True
@@ -99,7 +103,7 @@ def run_record_ide(
99103

100104

101105
def cancel_container(name: str, container: Any):
102-
TIMEOUT = 15
106+
TIMEOUT = 60
103107
time.sleep(TIMEOUT)
104108
if name in __containers_running and __containers_running[name]:
105109
__containers_running[name] = False
@@ -133,9 +137,10 @@ def pack_record_data(project_id: str, record_id: str) -> str:
133137
pickle.dump(full_data, file)
134138
return record_bytes_path
135139

140+
136141
def pack_knowledge_base(project_id: str) -> str:
137142
knowledge_base_source = knowledge_base.build_knowledge_base_from_project(project_id)
138143
knowledge_base_path = f"{project_id}knowledge_base.p"
139144
with open(knowledge_base_path, "wb") as file:
140145
pickle.dump(knowledge_base_source, file)
141-
return knowledge_base_path
146+
return knowledge_base_path

0 commit comments

Comments
 (0)