Skip to content

Commit 330157e

Browse files
fix(endpoints): wait flag should make it wait (#270)
Co-authored-by: Blaine Kasten <[email protected]>
1 parent 1201470 commit 330157e

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/together/cli/api/endpoints.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -137,8 +137,7 @@ def endpoints(ctx: click.Context) -> None:
137137
help="Start endpoint in specified availability zone (e.g., us-central-4b)",
138138
)
139139
@click.option(
140-
"--wait",
141-
is_flag=True,
140+
"--wait/--no-wait",
142141
default=True,
143142
help="Wait for the endpoint to be ready after creation",
144143
)
@@ -284,7 +283,9 @@ def fetch_and_print_hardware_options(
284283
@endpoints.command()
285284
@click.argument("endpoint-id", required=True)
286285
@click.option(
287-
"--wait", is_flag=True, default=True, help="Wait for the endpoint to stop"
286+
"--wait/--no-wait",
287+
default=True,
288+
help="Wait for the endpoint to stop",
288289
)
289290
@click.pass_obj
290291
@handle_api_errors
@@ -307,7 +308,9 @@ def stop(client: Together, endpoint_id: str, wait: bool) -> None:
307308
@endpoints.command()
308309
@click.argument("endpoint-id", required=True)
309310
@click.option(
310-
"--wait", is_flag=True, default=True, help="Wait for the endpoint to start"
311+
"--wait/--no-wait",
312+
default=True,
313+
help="Wait for the endpoint to start",
311314
)
312315
@click.pass_obj
313316
@handle_api_errors

0 commit comments

Comments
 (0)