update tests

Author: pintaoz

hyperpod-cluster-stack-template/hyperpod_cluster_stack_template/v1_0/model.py

@@ -51,7 +51,7 @@ class ClusterStackBase(BaseModel):
     file_system_type_version: Optional[float] = Field(2.15, description="File system type version for the FSx file system")
     storage_capacity: Optional[int] = Field(1200, description="Storage capacity for the FSx file system in GiB")
     fsx_file_system_id: Optional[str] = Field("", description="Existing FSx file system ID")
-    template_version: str = Field("1", description="Version number of cluster creation template")
+    template_version: Optional[str] = Field("1", description="Version number of cluster creation template")
 
     @field_validator('kubernetes_version', mode='before')
     @classmethod

test/integration_tests/cluster_management/test_hp_cluster_creation.py

@@ -148,7 +148,8 @@ 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"
+        "create-fsx-stack": "false",
+        "template-version": "1"
     }
 
     # Build CLI arguments
@@ -170,7 +171,8 @@ 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
+        "create_fsx_stack": False,
+        "template-version": "1"
     }
     assert_config_values("./", expected_config)