Add --instance flag to null pyworker client

Lets the demo target run-alpha.vast.ai (or candidate/local) without
editing code. Defaults to prod; respects VAST_INSTANCE env var.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Rob Ballantyne
2026-05-12 11:40:51 +01:00
parent 34fd21e76a
commit 47ad0ebe0a
+8 -2
View File
@@ -169,6 +169,12 @@ def build_arg_parser() -> argparse.ArgumentParser:
f"session lands without queueing (default: {DEFAULT_SESSION_COST})" f"session lands without queueing (default: {DEFAULT_SESSION_COST})"
), ),
) )
p.add_argument(
"--instance",
choices=("prod", "alpha", "candidate", "local"),
default=os.environ.get("VAST_INSTANCE", "prod"),
help="Vast serverless instance to target (default: prod)",
)
return p return p
@@ -176,11 +182,11 @@ async def main_async():
args = build_arg_parser().parse_args() args = build_arg_parser().parse_args()
print("=" * 60) print("=" * 60)
print(f"Endpoint: {args.endpoint}") print(f"Endpoint: {args.endpoint} (instance: {args.instance})")
print("=" * 60) print("=" * 60)
try: try:
async with Serverless() as client: async with Serverless(instance=args.instance) as client:
if args.demo: if args.demo:
await run_demo( await run_demo(
client, client,