minor

skypilot-org · Jun 8, 2024 · 9dca996 · 9dca996
1 parent 2922de0
commit 9dca996
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/sky/api/sdk.py b/sky/api/sdk.py
@@ -61,6 +61,7 @@ def _start_uvicorn_in_background():
     # Wait for the server to start.
     while True:
         try:
+            # TODO: Should check the process is running as well.
             requests.get(f'{_get_server_url()}/health', timeout=1)
             break
         except requests.exceptions.ConnectionError:

diff --git a/sky/setup_files/setup.py b/sky/setup_files/setup.py
@@ -162,6 +162,7 @@ def parse_readme(readme: str) -> str:
     # <= 3.13 may encounter https://github.com/ultralytics/yolov5/issues/414
     'pyyaml > 3.13, != 5.4.*',
     'requests',
+    'fastapi'
 ]
 
 local_ray = [