update init files (#298)

Author: ShilinHe

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"],

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"]