Skip to content

Commit 2a9ab73

Browse files
author
Dan Choi
committed
change local_model_path to local_code_path
1 parent 1e896ec commit 2a9ab73

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,7 @@ def repack_model(inference_script, source_directory, model_uri, sagemaker_sessio
344344
local_code_path = os.path.join(tmp, 'local_code.tar.gz')
345345
download_file_from_url(source_directory, local_code_path, sagemaker_session)
346346

347-
with tarfile.open(name=local_model_path, mode='r:gz') as t:
347+
with tarfile.open(name=local_code_path, mode='r:gz') as t:
348348
t.extractall(path=code_dir)
349349

350350
elif source_directory:

0 commit comments

Comments
 (0)