Skip to content

Commit 79ca9d4

Browse files
author
pintaoz
committed
update default_create
1 parent 3fffde3 commit 79ca9d4

File tree

2 files changed

+4
-3
lines changed
  • hyperpod-cluster-stack-template/hyperpod_cluster_stack_template/v1_0
  • src/sagemaker/hyperpod/cli/commands

2 files changed

+4
-3
lines changed

hyperpod-cluster-stack-template/hyperpod_cluster_stack_template/v1_0/schema.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -440,7 +440,7 @@
440440
},
441441
"custom_bucket_name": {
442442
"default": "",
443-
"description": "S3 bucket name for templates",
443+
"description": "Custom S3 bucket name for templates",
444444
"title": "Custom Bucket Name",
445445
"type": "string"
446446
},

src/sagemaker/hyperpod/cli/commands/init.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -271,8 +271,9 @@ def validate():
271271

272272
@click.command(name="_default_create")
273273
@click.option("--region", "-r", default=None, help="Region to create cluster stack for, default to your region in aws configure. Not available for other templates.")
274+
@click.option("--template-version", type=click.INT, help="Version number of cluster creation template. Not available for other templates.")
274275
@_hyperpod_telemetry_emitter(Feature.HYPERPOD_CLI, "init_create_cli")
275-
def _default_create(region):
276+
def _default_create(region, template_version):
276277
"""
277278
Validate configuration and render template files for deployment.
278279
@@ -373,7 +374,7 @@ def _default_create(region):
373374
# Pass region to to_domain for cluster stack template
374375
if template == "cluster-stack":
375376
config = flat.to_config(region=region)
376-
HpClusterStack(**config).create(region)
377+
HpClusterStack(**config).create(region, template_version)
377378
else:
378379
domain = flat.to_domain()
379380
domain.create()

0 commit comments

Comments
 (0)