Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
pintaoz-aws committed Jan 13, 2025
1 parent 9e70fa3 commit 12937f3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/sagemaker/modules/local_core/local_container.py
Original file line number Diff line number Diff line change
Expand Up @@ -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."""
Expand Down Expand Up @@ -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

Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 12937f3

Please sign in to comment.