Update tests

Author: pintaoz

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

@@ -53,7 +53,7 @@ def parse_status_list(ctx, param, value):
 @click.argument("config-file", required=True)
 @click.argument("stack-name", required=True)
 @click.option("--region", help="AWS region")
-@click.option("--template-version", help="Version number of cluster creation template")
+@click.option("--template-version", type=click.INT, help="Version number of cluster creation template")
 @click.option("--debug", is_flag=True, help="Enable debug logging")
 def create_cluster_stack(config_file, region, template_version, debug):
     """Create a new HyperPod cluster stack using the provided configuration.

test/integration_tests/cluster_management/test_hp_cluster_creation.py

@@ -148,8 +148,7 @@ def test_configure_cluster(runner, cluster_name):
         "create-sagemaker-iam-role-stack": "true",
         "create-hyperpod-cluster-stack": "true",
         "create-helm-chart-stack": "true",
-        "create-fsx-stack": "false",
-        "template-version": "1"
+        "create-fsx-stack": "false"
     }
 
     # Build CLI arguments
@@ -171,8 +170,7 @@ def test_configure_cluster(runner, cluster_name):
         "create_sagemaker_iam_role_stack": True,
         "create_hyperpod_cluster_stack": True,
         "create_helm_chart_stack": True,
-        "create_fsx_stack": False,
-        "template-version": "1"
+        "create_fsx_stack": False
     }
     assert_config_values("./", expected_config)
 
@@ -192,7 +190,7 @@ def test_create_cluster(runner, cluster_name, create_time):
     # Record time before submission
     CREATE_TIME = datetime.now(timezone.utc)
 
-    result = runner.invoke(create, ["--region", REGION], catch_exceptions=False)
+    result = runner.invoke(create, ["--region", REGION, "--template-version", "1"], catch_exceptions=False)
     assert_command_succeeded(result)
 
     # Verify expected submission messages appear