File tree 2 files changed +4
-4
lines changed
src/sagemaker/remote_function
tests/unit/sagemaker/remote_function
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 56
56
RUNTIME_MANAGER_SCRIPT_NAME = "runtime_environment_manager.py"
57
57
58
58
# training channel names
59
- RUNTIME_SCRIPTS_CHANNEL_NAME = "sm_rf_bootstrap "
59
+ RUNTIME_SCRIPTS_CHANNEL_NAME = "sagemaker_remote_function_bootstrap "
60
60
REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
61
61
SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
62
62
Original file line number Diff line number Diff line change 39
39
KMS_KEY_ARN = "kms-key-arn"
40
40
DEFAULT_ROLE_ARN = "default_execution_role_arn"
41
41
TEST_REGION = "us-west-2"
42
- RUNTIME_SCRIPTS_CHANNEL_NAME = "sm_rf_bootstrap "
42
+ RUNTIME_SCRIPTS_CHANNEL_NAME = "sagemaker_remote_function_bootstrap "
43
43
REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
44
44
SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
45
45
SAGEMAKER_SDK_WHL_FILE = (
@@ -283,7 +283,7 @@ def test_start(
283
283
TrainingInputMode = "File" ,
284
284
ContainerEntrypoint = [
285
285
"/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" ,
287
287
],
288
288
ContainerArguments = [
289
289
"--s3_base_uri" ,
@@ -400,7 +400,7 @@ def test_start_with_complete_job_settings(
400
400
TrainingInputMode = "File" ,
401
401
ContainerEntrypoint = [
402
402
"/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" ,
404
404
],
405
405
ContainerArguments = [
406
406
"--s3_base_uri" ,
You can’t perform that action at this time.
0 commit comments