@@ -108,7 +108,7 @@ class _LocalContainer(BaseModel):
108
108
container_entrypoint : Optional [List [str ]]
109
109
container_arguments : Optional [List [str ]]
110
110
111
- _temperary_folders : List [str ] = []
111
+ _temporary_folders : List [str ] = []
112
112
113
113
def model_post_init (self , __context : Any ):
114
114
"""Post init method to perform custom validation and set default values."""
@@ -212,7 +212,7 @@ def train(
212
212
shutil .rmtree (os .path .join (self .container_root , "shared" ))
213
213
for host in self .hosts :
214
214
shutil .rmtree (os .path .join (self .container_root , host ))
215
- for folder in self ._temperary_folders :
215
+ for folder in self ._temporary_folders :
216
216
shutil .rmtree (os .path .join (self .container_root , folder ))
217
217
return artifacts
218
218
@@ -553,7 +553,7 @@ def _get_data_source_local_path(self, data_source: DataSource):
553
553
uri = data_source .s3_data_source .s3_uri
554
554
parsed_uri = urlparse (uri )
555
555
local_dir = TemporaryDirectory (prefix = os .path .join (self .container_root + "/" )).name
556
- self ._temperary_folders .append (local_dir )
556
+ self ._temporary_folders .append (local_dir )
557
557
download_folder (parsed_uri .netloc , parsed_uri .path , local_dir , self .sagemaker_session )
558
558
return local_dir
559
559
else :
0 commit comments