gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 07:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: move docker timeout message to __run() (#620)

Browse Source 
* move docker timeout msg to __run()

* fix CI
This commit is contained in:
XianBW
2025-02-19 19:09:52 +08:00
committed by GitHub
parent 05c7712770
commit fc5f295de7
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/utils/env.py
+8 -6
View File
@@ -366,6 +366,7 @@ class DockerEnv(Env[DockerConf]):
log_output = ""
try:
start = time.time()
container: docker.models.containers.Container = client.containers.run( # type: ignore[no-any-unimported]
image=self.conf.image,
command=entry,
@@ -396,10 +397,16 @@ class DockerEnv(Env[DockerConf]):
decoded_log = self.replace_time_info(decoded_log) if remove_timestamp else decoded_log
Console().print(decoded_log, markup=False)
log_output += decoded_log + "\n"
print(Rule("[bold green]Docker Logs End[/bold green]", style="dark_orange"))
container.wait()
container.stop()
container.remove()
end = time.time()
if end - start >= self.conf.running_timeout_period:
print(
f"[red]The running time exceeds {self.conf.running_timeout_period} seconds, so the process is killed.[/red]"
)
log_output += f"\n\nThe running time exceeds {self.conf.running_timeout_period} seconds, so the process is killed."
print(Rule("[bold green]Docker Logs End[/bold green]", style="dark_orange"))
return log_output
except docker.errors.ContainerError as e:
raise RuntimeError(f"Error while running the container: {e}")
@@ -509,17 +516,12 @@ class DockerEnv(Env[DockerConf]):
f"/bin/sh -c 'timeout {self.conf.running_timeout_period} {entry}; chmod -R 777 {self.conf.mount_path}'"
)
start = time.time()
if self.conf.enable_cache:
out = self.cached_run(entry_add_timeout, local_path, env, running_extra_volume)
else:
out = self.__run_with_retry(
entry_add_timeout, local_path, env, running_extra_volume, remove_timestamp=False
)
end = time.time()
if end - start + 1 >= self.conf.running_timeout_period:
out += f"\n\nThe running time exceeds {self.conf.running_timeout_period} seconds, so the process is killed."
return out