update init files (#298)

microsoft · Apr 11, 2024 · a7569bf · a7569bf
1 parent ce9e433
commit a7569bf
Show file tree

Hide file tree

Showing 10 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -76,7 +76,7 @@ def revert():
             "Operating System :: OS Independent",
         ],
         package_data={
-            "taskweaver": ["**/*.yaml"],
+            "taskweaver": ["**/*.yaml", "**/*.yml"],
         },
         entry_points={
             "console_scripts": ["taskweaver=taskweaver.__main__:main"],

diff --git a/taskweaver/ces/environment.py b/taskweaver/ces/environment.py
@@ -259,15 +259,15 @@ def start_session(
             )
 
             tick = 0
-            while tick < 10:
+            while tick < 30:
                 container.reload()
                 if container.status == "running" and os.path.isfile(connection_file):
                     logger.info("Container is running and connection file is ready.")
                     break
                 time.sleep(1)  # wait for 1 second before checking again
                 tick += 1
-            if tick == 10:
-                raise Exception("Container is not ready after 10 seconds")
+            if tick == 30:
+                raise Exception("Container is not ready after 30 seconds")
 
             # save the ports to ces session dir
             port_bindings = container.attrs["NetworkSettings"]["Ports"]

diff --git a/taskweaver/ces/kernel/__init__.py b/taskweaver/ces/kernel/__init__.py
diff --git a/taskweaver/ces/manager/__init__.py b/taskweaver/ces/manager/__init__.py
diff --git a/taskweaver/ces/runtime/__init__.py b/taskweaver/ces/runtime/__init__.py
diff --git a/taskweaver/chat/__init__.py b/taskweaver/chat/__init__.py
diff --git a/taskweaver/ext_role/echo/__init__.py b/taskweaver/ext_role/echo/__init__.py
diff --git a/taskweaver/ext_role/web_explorer/__init__.py b/taskweaver/ext_role/web_explorer/__init__.py
diff --git a/taskweaver/ext_role/web_search/__init__.py b/taskweaver/ext_role/web_search/__init__.py
diff --git a/taskweaver/module/__init__.py b/taskweaver/module/__init__.py