Skip to content

Commit 56b1291

Browse files
bmouryakrMourya Baddam
authored and
Namrata Madan
committed
change: rename container entrypoint to identify pathways jobs (aws#906)
Co-authored-by: Mourya Baddam <[email protected]>
1 parent 6418ead commit 56b1291

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/sagemaker/remote_function/job.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
RUNTIME_MANAGER_SCRIPT_NAME = "runtime_environment_manager.py"
5757

5858
# training channel names
59-
RUNTIME_SCRIPTS_CHANNEL_NAME = "sm_rf_bootstrap"
59+
RUNTIME_SCRIPTS_CHANNEL_NAME = "sagemaker_remote_function_bootstrap"
6060
REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
6161
SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
6262

tests/unit/sagemaker/remote_function/test_job.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
KMS_KEY_ARN = "kms-key-arn"
4040
DEFAULT_ROLE_ARN = "default_execution_role_arn"
4141
TEST_REGION = "us-west-2"
42-
RUNTIME_SCRIPTS_CHANNEL_NAME = "sm_rf_bootstrap"
42+
RUNTIME_SCRIPTS_CHANNEL_NAME = "sagemaker_remote_function_bootstrap"
4343
REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
4444
SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
4545
SAGEMAKER_SDK_WHL_FILE = (
@@ -283,7 +283,7 @@ def test_start(
283283
TrainingInputMode="File",
284284
ContainerEntrypoint=[
285285
"/bin/bash",
286-
"/opt/ml/input/data/sm_rf_bootstrap/job_driver.sh",
286+
"/opt/ml/input/data/sagemaker_remote_function_bootstrap/job_driver.sh",
287287
],
288288
ContainerArguments=[
289289
"--s3_base_uri",
@@ -400,7 +400,7 @@ def test_start_with_complete_job_settings(
400400
TrainingInputMode="File",
401401
ContainerEntrypoint=[
402402
"/bin/bash",
403-
"/opt/ml/input/data/sm_rf_bootstrap/job_driver.sh",
403+
"/opt/ml/input/data/sagemaker_remote_function_bootstrap/job_driver.sh",
404404
],
405405
ContainerArguments=[
406406
"--s3_base_uri",

0 commit comments

Comments
 (0)