Update resources_codegen based on sagemaker-core change

Author: mollyheamazon

sagemaker-core/src/sagemaker/core/tools/resources_codegen.py

@@ -1664,12 +1664,19 @@ def _get_instance_count_ref(self, resource_name: str) -> str:
         """
 
         if resource_name == "TrainingJob":
-            return """(
-                sum(instance_group.instance_count for instance_group in self.resource_config.instance_groups)
-                if self.resource_config.instance_groups and not isinstance(self.resource_config.instance_groups, Unassigned)
-                else self.resource_config.instance_count
-            )
-            """
+            return """1  # Default
+        if not isinstance(self.resource_config, Unassigned):
+            if (
+                hasattr(self.resource_config, "instance_groups")
+                and self.resource_config.instance_groups
+                and not isinstance(self.resource_config.instance_groups, Unassigned)
+            ):
+                instance_count = sum(
+                    instance_group.instance_count
+                    for instance_group in self.resource_config.instance_groups
+                )
+            elif hasattr(self.resource_config, "instance_count"):
+                instance_count = self.resource_config.instance_count"""
         elif resource_name == "TransformJob":
             return "self.transform_resources.instance_count"
         elif resource_name == "ProcessingJob":

sagemaker-core/src/sagemaker/core/tools/templates.py

@@ -483,7 +483,7 @@ def delete(
 @Base.add_validate_call
 def stop(self) -> None:
 {docstring}
-    client = SageMakerClient().client
+    client = SageMakerClient().sagemaker_client
 
     operation_input_args = {{
 {operation_input_args}