Skip to content

Commit 488d24e

Browse files
committed
Remove remaining REMOTE_FUNCTION_SECRET_KEY references from tests
1 parent 55abc74 commit 488d24e

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

sagemaker-core/tests/unit/remote_function/test_invoke_function.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,6 @@ class TestMain:
224224
@patch("sagemaker.core.remote_function.invoke_function._get_sagemaker_session")
225225
@patch("sagemaker.core.remote_function.invoke_function._load_pipeline_context")
226226
@patch("sagemaker.core.remote_function.invoke_function._parse_args")
227-
@patch.dict("os.environ", {"REMOTE_FUNCTION_SECRET_KEY": "test-key"})
228227
def test_main_success(self, mock_parse, mock_load_context, mock_get_session, mock_execute):
229228
"""Test main function successful execution."""
230229
mock_args = MagicMock()
@@ -252,7 +251,6 @@ def test_main_success(self, mock_parse, mock_load_context, mock_get_session, moc
252251
@patch("sagemaker.core.remote_function.invoke_function._get_sagemaker_session")
253252
@patch("sagemaker.core.remote_function.invoke_function._load_pipeline_context")
254253
@patch("sagemaker.core.remote_function.invoke_function._parse_args")
255-
@patch.dict("os.environ", {"REMOTE_FUNCTION_SECRET_KEY": "test-key"})
256254
def test_main_handles_exception(
257255
self, mock_parse, mock_load_context, mock_get_session, mock_execute, mock_handle_error
258256
):

sagemaker-core/tests/unit/remote_function/test_job.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -875,7 +875,7 @@ def test_start(self, mock_get_name, mock_compile, mock_session):
875875
mock_get_name.return_value = "test-job"
876876
mock_compile.return_value = {
877877
"TrainingJobName": "test-job",
878-
"Environment": {"REMOTE_FUNCTION_SECRET_KEY": "test-key"},
878+
"Environment": {},
879879
}
880880

881881
job_settings = Mock()

0 commit comments

Comments
 (0)