@@ -1177,8 +1177,14 @@ def test_retry_with_backoff(patched_sleep):
1177
1177
assert retry_with_backoff (callable_func , 2 ) == func_return_val
1178
1178
1179
1179
1180
- @patch ("sagemaker.utils.logger.info" )
1181
- def test_resolve_value_from_config (mock_info_logger ):
1180
+ @patch ("sagemaker.utils.get_sagemaker_config_logger" )
1181
+ def test_resolve_value_from_config (mock_get_sagemaker_config_logger ):
1182
+
1183
+ mock_config_logger = Mock ()
1184
+ mock_get_sagemaker_config_logger .return_value = mock_config_logger
1185
+
1186
+ mock_info_logger = Mock ()
1187
+ mock_config_logger .info = mock_info_logger
1182
1188
# using a shorter name for inside the test
1183
1189
sagemaker_session = MagicMock ()
1184
1190
sagemaker_session .settings .ignore_sagemaker_config = False
@@ -1235,12 +1241,16 @@ def test_resolve_value_from_config(mock_info_logger):
1235
1241
1236
1242
sagemaker_session .settings .ignore_sagemaker_config = True
1237
1243
assert resolve_value_from_config ("blah" , config_key_path , None , sagemaker_session ) == "blah"
1238
- mock_info_logger .assert_called_once_with ("Ignoring defaults config." )
1244
+ mock_info_logger .assert_called_once_with (
1245
+ "Ignoring defaults config because `ignore_sagemaker_config` was set in Session settings."
1246
+ )
1239
1247
1240
1248
mock_info_logger .reset_mock ()
1241
1249
1242
1250
assert resolve_value_from_config (None , config_key_path , "blah" , sagemaker_session ) == "blah"
1243
- mock_info_logger .assert_called_once_with ("Ignoring defaults config." )
1251
+ mock_info_logger .assert_called_once_with (
1252
+ "Ignoring defaults config because `ignore_sagemaker_config` was set in Session settings."
1253
+ )
1244
1254
1245
1255
1246
1256
@patch ("jsonschema.validate" )
0 commit comments