From eaf1b62343f1aade858ddc82255b7d9380ee51ef Mon Sep 17 00:00:00 2001
From: Yizhou Chi <chiyizhou@fuzhi.ai>
Date: Wed, 11 Sep 2024 16:14:50 +0800
Subject: [PATCH] add timeout to client

---
 expo/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/expo/utils.py b/expo/utils.py
index 44de8cf9b..f3c0c392d 100644
--- a/expo/utils.py
+++ b/expo/utils.py
@@ -111,7 +111,7 @@ async def load_execute_notebook(role):
     codes = [task.code for task in tasks if task.code]
     executor = role.execute_code
     executor.nb = nbformat.v4.new_notebook()
-    executor.nb_client = NotebookClient(executor.nb)
+    executor.nb_client = NotebookClient(executor.nb, timeout=executor.timeout)
     # await executor.build()
     for code in codes:
         outputs, success = await executor.run(code)