sagemaker-core 1.0.9__py3-none-any.whl → 1.0.11__py3-none-any.whl

sagemaker_core/main/resources.py

@@ -41,6 +41,7 @@ from sagemaker_core.main.intelligent_defaults_helper import (
     load_default_configs_for_resource_name,
     get_config_value,
 )
+from sagemaker_core.main.logs import MultiLogStreamHandler
 from sagemaker_core.main.shapes import *
 from sagemaker_core.main.exceptions import *
 
@@ -812,7 +813,8 @@ class Algorithm(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -977,6 +979,7 @@ class App(Base):
         creation_time: The creation time of the application.  After an application has been shut down for 24 hours, SageMaker deletes all metadata for the application. To be considered an update and retain application metadata, applications must be restarted within 24 hours after the previous application has been shut down. After this time window, creation of an application is considered a new application rather than an update of the previous application.
         failure_reason: The failure reason.
         resource_spec: The instance type and the Amazon Resource Name (ARN) of the SageMaker image created on the instance.
+        built_in_lifecycle_config_arn: The lifecycle configuration that runs before the default lifecycle configuration
 
     """
 
@@ -992,6 +995,7 @@ class App(Base):
     creation_time: Optional[datetime.datetime] = Unassigned()
     failure_reason: Optional[str] = Unassigned()
     resource_spec: Optional[ResourceSpec] = Unassigned()
+    built_in_lifecycle_config_arn: Optional[str] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)
@@ -1270,7 +1274,8 @@ class App(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -2652,7 +2657,11 @@ class AutoMLJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a AutoMLJob resource.
 
@@ -2682,7 +2691,8 @@ class AutoMLJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -3130,7 +3140,11 @@ class AutoMLJobV2(Base):
         return self
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a AutoMLJobV2 resource.
 
@@ -3160,7 +3174,8 @@ class AutoMLJobV2(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -3534,7 +3549,8 @@ class Cluster(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -4493,7 +4509,11 @@ class CompilationJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a CompilationJob resource.
 
@@ -4523,7 +4543,8 @@ class CompilationJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -6148,7 +6169,7 @@ class Domain(Base):
         kms_key_id: The Amazon Web Services KMS customer managed key used to encrypt the EFS volume attached to the domain.
         app_security_group_management: The entity that creates and manages the required security groups for inter-app communication in VPCOnly mode. Required when CreateDomain.AppNetworkAccessType is VPCOnly and DomainSettings.RStudioServerProDomainSettings.DomainExecutionRoleArn is provided.
         tag_propagation: Indicates whether custom tag propagation is supported for the domain.
-        default_space_settings: The default settings used to create a space.
+        default_space_settings: The default settings for shared spaces that users create in the domain.
 
     """
 
@@ -6293,7 +6314,7 @@ class Domain(Base):
             kms_key_id: SageMaker uses Amazon Web Services KMS to encrypt EFS and EBS volumes attached to the domain with an Amazon Web Services managed key by default. For more control, specify a customer managed key.
             app_security_group_management: The entity that creates and manages the required security groups for inter-app communication in VPCOnly mode. Required when CreateDomain.AppNetworkAccessType is VPCOnly and DomainSettings.RStudioServerProDomainSettings.DomainExecutionRoleArn is provided. If setting up the domain for use with RStudio, this value must be set to Service.
             tag_propagation: Indicates whether custom tag propagation is supported for the domain. Defaults to DISABLED.
-            default_space_settings: The default settings used to create a space.
+            default_space_settings: The default settings for shared spaces that users create in the domain.
             session: Boto3 session.
             region: Region name.
 
@@ -6581,7 +6602,8 @@ class Domain(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -7530,7 +7552,11 @@ class EdgePackagingJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a EdgePackagingJob resource.
 
@@ -7560,7 +7586,8 @@ class EdgePackagingJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -8031,7 +8058,8 @@ class Endpoint(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -9547,7 +9575,8 @@ class FeatureGroup(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -10415,7 +10444,8 @@ class FlowDefinition(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -10910,7 +10940,8 @@ class Hub(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -11299,7 +11330,8 @@ class HubContent(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -11874,7 +11906,8 @@ class HumanTaskUi(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -12341,7 +12374,11 @@ class HyperParameterTuningJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a HyperParameterTuningJob resource.
 
@@ -12371,7 +12408,8 @@ class HyperParameterTuningJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -12943,7 +12981,8 @@ class Image(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -13549,7 +13588,8 @@ class ImageVersion(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -13966,7 +14006,8 @@ class InferenceComponent(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -14601,7 +14642,8 @@ class InferenceExperiment(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -14974,7 +15016,11 @@ class InferenceRecommendationsJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a InferenceRecommendationsJob resource.
 
@@ -15004,7 +15050,8 @@ class InferenceRecommendationsJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -15536,7 +15583,11 @@ class LabelingJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a LabelingJob resource.
 
@@ -15566,7 +15617,8 @@ class LabelingJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -16314,7 +16366,8 @@ class MlflowTrackingServer(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -17572,7 +17625,8 @@ class ModelCard(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -17946,7 +18000,11 @@ class ModelCardExportJob(Base):
         return self
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a ModelCardExportJob resource.
 
@@ -17976,7 +18034,8 @@ class ModelCardExportJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -18933,7 +18992,8 @@ class ModelPackage(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -19400,7 +19460,8 @@ class ModelPackageGroup(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -20789,7 +20850,8 @@ class MonitoringSchedule(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -20919,7 +20981,7 @@ class NotebookInstance(Base):
         notebook_instance_lifecycle_config_name: Returns the name of a notebook instance lifecycle configuration. For information about notebook instance lifestyle configurations, see Step 2.1: (Optional) Customize a Notebook Instance
         direct_internet_access: Describes whether SageMaker provides internet access to the notebook instance. If this value is set to Disabled, the notebook instance does not have internet access, and cannot connect to SageMaker training and endpoint services. For more information, see Notebook Instances Are Internet-Enabled by Default.
         volume_size_in_gb: The size, in GB, of the ML storage volume attached to the notebook instance.
-        accelerator_types: A list of the Elastic Inference (EI) instance types associated with this notebook instance. Currently only one EI instance type can be associated with a notebook instance. For more information, see Using Elastic Inference in Amazon SageMaker.
+        accelerator_types: This parameter is no longer supported. Elastic Inference (EI) is no longer available. This parameter was used to specify a list of the EI instance types associated with this notebook instance.
         default_code_repository: The Git repository associated with the notebook instance as its default code repository. This can be either the name of a Git repository stored as a resource in your account, or the URL of a Git repository in Amazon Web Services CodeCommit or in any other Git repository. When you open a notebook instance, it opens in the directory that contains this repository. For more information, see Associating Git Repositories with SageMaker Notebook Instances.
         additional_code_repositories: An array of up to three Git repositories associated with the notebook instance. These can be either the names of Git repositories stored as resources in your account, or the URL of Git repositories in Amazon Web Services CodeCommit or in any other Git repository. These repositories are cloned at the same level as the default repository of your notebook instance. For more information, see Associating Git Repositories with SageMaker Notebook Instances.
         root_access: Whether root access is enabled or disabled for users of the notebook instance.  Lifecycle configurations need root access to be able to set up a notebook instance. Because of this, lifecycle configurations associated with a notebook instance always run with root access even if you disable root access for users.
@@ -21027,7 +21089,7 @@ class NotebookInstance(Base):
             lifecycle_config_name: The name of a lifecycle configuration to associate with the notebook instance. For information about lifestyle configurations, see Step 2.1: (Optional) Customize a Notebook Instance.
             direct_internet_access: Sets whether SageMaker provides internet access to the notebook instance. If you set this to Disabled this notebook instance is able to access resources only in your VPC, and is not be able to connect to SageMaker training and endpoint services unless you configure a NAT Gateway in your VPC. For more information, see Notebook Instances Are Internet-Enabled by Default. You can set the value of this parameter to Disabled only if you set a value for the SubnetId parameter.
             volume_size_in_gb: The size, in GB, of the ML storage volume to attach to the notebook instance. The default value is 5 GB.
-            accelerator_types: A list of Elastic Inference (EI) instance types to associate with this notebook instance. Currently, only one instance type can be associated with a notebook instance. For more information, see Using Elastic Inference in Amazon SageMaker.
+            accelerator_types: This parameter is no longer supported. Elastic Inference (EI) is no longer available. This parameter was used to specify a list of EI instance types to associate with this notebook instance.
             default_code_repository: A Git repository to associate with the notebook instance as its default code repository. This can be either the name of a Git repository stored as a resource in your account, or the URL of a Git repository in Amazon Web Services CodeCommit or in any other Git repository. When you open a notebook instance, it opens in the directory that contains this repository. For more information, see Associating Git Repositories with SageMaker Notebook Instances.
             additional_code_repositories: An array of up to three Git repositories to associate with the notebook instance. These can be either the names of Git repositories stored as resources in your account, or the URL of Git repositories in Amazon Web Services CodeCommit or in any other Git repository. These repositories are cloned at the same level as the default repository of your notebook instance. For more information, see Associating Git Repositories with SageMaker Notebook Instances.
             root_access: Whether root access is enabled or disabled for users of the notebook instance. The default value is Enabled.  Lifecycle configurations need root access to be able to set up a notebook instance. Because of this, lifecycle configurations associated with a notebook instance always run with root access even if you disable root access for users.
@@ -21208,7 +21270,7 @@ class NotebookInstance(Base):
         Parameters:
             lifecycle_config_name: The name of a lifecycle configuration to associate with the notebook instance. For information about lifestyle configurations, see Step 2.1: (Optional) Customize a Notebook Instance.
             disassociate_lifecycle_config: Set to true to remove the notebook instance lifecycle configuration currently associated with the notebook instance. This operation is idempotent. If you specify a lifecycle configuration that is not associated with the notebook instance when you call this method, it does not throw an error.
-            disassociate_accelerator_types: A list of the Elastic Inference (EI) instance types to remove from this notebook instance. This operation is idempotent. If you specify an accelerator type that is not associated with the notebook instance when you call this method, it does not throw an error.
+            disassociate_accelerator_types: This parameter is no longer supported. Elastic Inference (EI) is no longer available. This parameter was used to specify a list of the EI instance types to remove from this notebook instance.
             disassociate_default_code_repository: The name or URL of the default Git repository to remove from this notebook instance. This operation is idempotent. If you specify a Git repository that is not associated with the notebook instance when you call this method, it does not throw an error.
             disassociate_additional_code_repositories: A list of names or URLs of the default Git repositories to remove from this notebook instance. This operation is idempotent. If you specify a Git repository that is not associated with the notebook instance when you call this method, it does not throw an error.
 
@@ -21358,7 +21420,8 @@ class NotebookInstance(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -22197,7 +22260,11 @@ class OptimizationJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+    ) -> None:
         """
         Wait for a OptimizationJob resource.
 
@@ -22227,7 +22294,8 @@ class OptimizationJob(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -22698,7 +22766,8 @@ class Pipeline(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -23096,7 +23165,8 @@ class PipelineExecution(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -24073,13 +24143,19 @@ class ProcessingJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+        logs: Optional[bool] = False,
+    ) -> None:
         """
         Wait for a ProcessingJob resource.
 
         Parameters:
             poll: The number of seconds to wait between each poll.
             timeout: The maximum number of seconds to wait before timing out.
+            logs: Whether to print logs while waiting.
 
         Raises:
             TimeoutExceededError:  If the resource does not reach a terminal state before the timeout.
@@ -24098,18 +24174,32 @@ class ProcessingJob(Base):
         progress.add_task("Waiting for ProcessingJob...")
         status = Status("Current status:")
 
+        instance_count = self.processing_resources.cluster_config.instance_count
+        if logs:
+            multi_stream_logger = MultiLogStreamHandler(
+                log_group_name=f"/aws/sagemaker/ProcessingJobs",
+                log_stream_name_prefix=self.get_name(),
+                expected_stream_count=instance_count,
+            )
+
         with Live(
             Panel(
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
                 current_status = self.processing_job_status
                 status.update(f"Current status: [bold]{current_status}")
 
+                if logs and multi_stream_logger.ready():
+                    stream_log_events = multi_stream_logger.get_latest_log_events()
+                    for stream_id, event in stream_log_events:
+                        logger.info(f"{stream_id}:\n{event['message']}")
+
                 if current_status in terminal_states:
                     logger.info(f"Final Resource Status: [bold]{current_status}")
 
@@ -24537,7 +24627,8 @@ class Project(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -25182,7 +25273,8 @@ class Space(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -26433,13 +26525,19 @@ class TrainingJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+        logs: Optional[bool] = False,
+    ) -> None:
         """
         Wait for a TrainingJob resource.
 
         Parameters:
             poll: The number of seconds to wait between each poll.
             timeout: The maximum number of seconds to wait before timing out.
+            logs: Whether to print logs while waiting.
 
         Raises:
             TimeoutExceededError:  If the resource does not reach a terminal state before the timeout.
@@ -26458,18 +26556,41 @@ class TrainingJob(Base):
         progress.add_task("Waiting for TrainingJob...")
         status = Status("Current status:")
 
+        instance_count = (
+            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
+        )
+
+        if logs:
+            multi_stream_logger = MultiLogStreamHandler(
+                log_group_name=f"/aws/sagemaker/TrainingJobs",
+                log_stream_name_prefix=self.get_name(),
+                expected_stream_count=instance_count,
+            )
+
         with Live(
             Panel(
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
                 current_status = self.training_job_status
                 status.update(f"Current status: [bold]{current_status}")
 
+                if logs and multi_stream_logger.ready():
+                    stream_log_events = multi_stream_logger.get_latest_log_events()
+                    for stream_id, event in stream_log_events:
+                        logger.info(f"{stream_id}:\n{event['message']}")
+
                 if current_status in terminal_states:
                     logger.info(f"Final Resource Status: [bold]{current_status}")
 
@@ -26884,13 +27005,19 @@ class TransformJob(Base):
         logger.info(f"Stopping {self.__class__.__name__} - {self.get_name()}")
 
     @Base.add_validate_call
-    def wait(self, poll: int = 5, timeout: Optional[int] = None) -> None:
+    def wait(
+        self,
+        poll: int = 5,
+        timeout: Optional[int] = None,
+        logs: Optional[bool] = False,
+    ) -> None:
         """
         Wait for a TransformJob resource.
 
         Parameters:
             poll: The number of seconds to wait between each poll.
             timeout: The maximum number of seconds to wait before timing out.
+            logs: Whether to print logs while waiting.
 
         Raises:
             TimeoutExceededError:  If the resource does not reach a terminal state before the timeout.
@@ -26909,18 +27036,32 @@ class TransformJob(Base):
         progress.add_task("Waiting for TransformJob...")
         status = Status("Current status:")
 
+        instance_count = self.transform_resources.instance_count
+        if logs:
+            multi_stream_logger = MultiLogStreamHandler(
+                log_group_name=f"/aws/sagemaker/TransformJobs",
+                log_stream_name_prefix=self.get_name(),
+                expected_stream_count=instance_count,
+            )
+
         with Live(
             Panel(
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
                 current_status = self.transform_job_status
                 status.update(f"Current status: [bold]{current_status}")
 
+                if logs and multi_stream_logger.ready():
+                    stream_log_events = multi_stream_logger.get_latest_log_events()
+                    for stream_id, event in stream_log_events:
+                        logger.info(f"{stream_id}:\n{event['message']}")
+
                 if current_status in terminal_states:
                     logger.info(f"Final Resource Status: [bold]{current_status}")
 
@@ -27736,7 +27877,8 @@ class TrialComponent(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -28397,7 +28539,8 @@ class UserProfile(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()
@@ -28882,7 +29025,8 @@ class Workforce(Base):
                 Group(progress, status),
                 title="Wait Log Panel",
                 border_style=Style(color=Color.BLUE.value),
-            )
+            ),
+            transient=True,
         ):
             while True:
                 self.refresh()