Skip to content

Commit 3cb127f

Browse files
authored
fix: Address incorrect logic for endpoint [command] --wait false logic (#189)
1 parent 9c17199 commit 3cb127f

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/together/lib/cli/api/endpoints.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,7 @@ def endpoints(ctx: click.Context) -> None:
128128
help="Start endpoint in specified availability zone (e.g., us-central-4b)",
129129
)
130130
@click.option(
131-
"--wait",
132-
is_flag=True,
131+
"--wait/--no-wait",
133132
default=True,
134133
help="Wait for the endpoint to be ready after creation",
135134
)
@@ -272,7 +271,7 @@ def fetch_and_print_hardware_options(client: Together, model: str | None, print_
272271

273272
@endpoints.command()
274273
@click.argument("endpoint-id", required=True)
275-
@click.option("--wait", is_flag=True, default=True, help="Wait for the endpoint to stop")
274+
@click.option("--wait/--no-wait", default=True, help="Wait for the endpoint to stop")
276275
@click.pass_obj
277276
@handle_api_errors
278277
def stop(client: Together, endpoint_id: str, wait: bool) -> None:
@@ -293,7 +292,7 @@ def stop(client: Together, endpoint_id: str, wait: bool) -> None:
293292

294293
@endpoints.command()
295294
@click.argument("endpoint-id", required=True)
296-
@click.option("--wait", is_flag=True, default=True, help="Wait for the endpoint to start")
295+
@click.option("--wait/--no-wait", default=True, help="Wait for the endpoint to start")
297296
@click.pass_obj
298297
@handle_api_errors
299298
def start(client: Together, endpoint_id: str, wait: bool) -> None:

0 commit comments

Comments
 (0)