diff --git a/src/sagemaker/modules/train/model_trainer.py b/src/sagemaker/modules/train/model_trainer.py index 4183fb87cd..96078d1aeb 100644 --- a/src/sagemaker/modules/train/model_trainer.py +++ b/src/sagemaker/modules/train/model_trainer.py @@ -865,7 +865,7 @@ def _prepare_train_script( working_dir = f"cd {SM_CODE_CONTAINER_PATH} \n" if source_code.source_dir.endswith(".tar.gz"): tarfile_name = os.path.basename(source_code.source_dir) - working_dir += f"tar --strip-components=1 -xzf {tarfile_name} \n" + working_dir += f"tar -xzf {tarfile_name} \n" if base_command: execute_driver = EXECUTE_BASE_COMMANDS.format(base_command=base_command) diff --git a/tests/data/modules/script_mode/code.tar.gz b/tests/data/modules/script_mode/code.tar.gz index 7c43f35f57..e2ed9d4b18 100644 Binary files a/tests/data/modules/script_mode/code.tar.gz and b/tests/data/modules/script_mode/code.tar.gz differ