File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
src/sagemaker/remote_function
tests/unit/sagemaker/remote_function Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 5656RUNTIME_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 "
6060REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
6161SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
6262
Original file line number Diff line number Diff line change 3939KMS_KEY_ARN = "kms-key-arn"
4040DEFAULT_ROLE_ARN = "default_execution_role_arn"
4141TEST_REGION = "us-west-2"
42- RUNTIME_SCRIPTS_CHANNEL_NAME = "sm_rf_bootstrap "
42+ RUNTIME_SCRIPTS_CHANNEL_NAME = "sagemaker_remote_function_bootstrap "
4343REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
4444SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
4545SAGEMAKER_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" ,
You can’t perform that action at this time.
0 commit comments