From 0144d88d2c9a1d4b0ed9a729215b27ab84413031 Mon Sep 17 00:00:00 2001 From: Zhanghao Wu Date: Sun, 2 Jun 2024 20:23:34 +0000 Subject: [PATCH] format --- sky/api/rest.py | 5 +++-- sky/api/sdk.py | 1 - 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/sky/api/rest.py b/sky/api/rest.py index b15a6cd35d1..c4ee3907520 100644 --- a/sky/api/rest.py +++ b/sky/api/rest.py @@ -63,12 +63,13 @@ async def refresh_cluster_status_event(): refresh_cluster_status_event, ] + class RequestBody(pydantic.BaseModel): env_vars: Dict[str, str] = {} -def wrapper(func: Callable[P, Any], request_id: str, env_vars: Dict[str, str], *args: P.args, - **kwargs: P.kwargs): +def wrapper(func: Callable[P, Any], request_id: str, env_vars: Dict[str, str], + *args: P.args, **kwargs: P.kwargs): """Wrapper for a request task.""" def redirect_output(file): diff --git a/sky/api/sdk.py b/sky/api/sdk.py index 97ac0b90244..41b41122ab8 100644 --- a/sky/api/sdk.py +++ b/sky/api/sdk.py @@ -107,7 +107,6 @@ def _add_env_vars_to_body(body: Dict[str, Any]): body['env_vars'] = env_vars - @usage_lib.entrypoint @_check_health def launch(