fix: filter with lambda

skypilot-org · Oct 1, 2024 · b824a59 · b824a59
1 parent 743a34e
commit b824a59
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/sky/jobs/state.py b/sky/jobs/state.py
@@ -22,8 +22,7 @@
 
 
 def _get_db_path() -> str:
-    """
-    Workaround to collapse multi-step Path ops for type checker.
+    """Workaround to collapse multi-step Path ops for type checker.
     Ensures _DB_PATH is str, avoiding Union[Path, str] inference.
     """
     path = pathlib.Path('~/.sky/spot_jobs.db')

diff --git a/sky/serve/serve_state.py b/sky/serve/serve_state.py
@@ -19,8 +19,7 @@
 
 
 def _get_db_path() -> str:
-    """
-    Workaround to collapse multi-step Path ops for type checker.
+    """Workaround to collapse multi-step Path ops for type checker.
     Ensures _DB_PATH is str, avoiding Union[Path, str] inference.
     """
     path = pathlib.Path(constants.SKYSERVE_METADATA_DIR) / 'services.db'

diff --git a/sky/serve/service.py b/sky/serve/service.py
@@ -241,8 +241,9 @@ def _get_host():
     finally:
         # Kill load balancer process first since it will raise errors if failed
         # to connect to the controller. Then the controller process.
-        process_to_kill = filter(lambda proc: proc is not None,
-                                 [load_balancer_process, controller_process])
+        process_to_kill = filter(None,
+                                 [load_balancer_process, controller_process
+                                 ])  # filter out None from the list
         subprocess_utils.kill_children_processes(
             [process.pid for process in process_to_kill], force=True)
         for process in process_to_kill: