From 12937f3113c2304275dbd5aca685470a81fec653 Mon Sep 17 00:00:00 2001 From: pintaoz-aws Date: Thu, 26 Dec 2024 18:07:50 -0800 Subject: [PATCH] fix typo --- src/sagemaker/modules/local_core/local_container.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/sagemaker/modules/local_core/local_container.py b/src/sagemaker/modules/local_core/local_container.py index e9709efb8f..3cfb52e626 100644 --- a/src/sagemaker/modules/local_core/local_container.py +++ b/src/sagemaker/modules/local_core/local_container.py @@ -108,7 +108,7 @@ class _LocalContainer(BaseModel): container_entrypoint: Optional[List[str]] container_arguments: Optional[List[str]] - _temperary_folders: List[str] = [] + _temporary_folders: List[str] = [] def model_post_init(self, __context: Any): """Post init method to perform custom validation and set default values.""" @@ -212,7 +212,7 @@ def train( shutil.rmtree(os.path.join(self.container_root, "shared")) for host in self.hosts: shutil.rmtree(os.path.join(self.container_root, host)) - for folder in self._temperary_folders: + for folder in self._temporary_folders: shutil.rmtree(os.path.join(self.container_root, folder)) return artifacts @@ -553,7 +553,7 @@ def _get_data_source_local_path(self, data_source: DataSource): uri = data_source.s3_data_source.s3_uri parsed_uri = urlparse(uri) local_dir = TemporaryDirectory(prefix=os.path.join(self.container_root + "/")).name - self._temperary_folders.append(local_dir) + self._temporary_folders.append(local_dir) download_folder(parsed_uri.netloc, parsed_uri.path, local_dir, self.sagemaker_session) return local_dir else: