Skip to content

Commit

Permalink
fix format
Browse files Browse the repository at this point in the history
  • Loading branch information
humpydonkey committed Jul 2, 2024
1 parent fb51bf7 commit 4a04acc
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions vision_agent/utils/execute.py
Original file line number Diff line number Diff line change
Expand Up @@ -446,8 +446,10 @@ def restart_kernel(self) -> None:
)
def exec_cell(self, code: str) -> Execution:
if not self.interpreter.is_running():
raise ConnectionResetError("Remote sandbox is not closed unexpectedly. Please retry the operation.")
self.interpreter.set_timeout(_SESSION_TIMEOUT) # Extend the life of the sandbox
raise ConnectionResetError(
"Remote sandbox is not closed unexpectedly. Please retry the operation."
)
self.interpreter.set_timeout(_SESSION_TIMEOUT) # Extend the life of the sandbox
execution = self.interpreter.notebook.exec_cell(code, timeout=self.timeout)
return Execution.from_e2b_execution(execution)

Expand Down

0 comments on commit 4a04acc

Please sign in to comment.