fix tests

aws · Jun 19, 2024 · b35b288 · b35b288
1 parent addd339
commit b35b288
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/unit/test_processing.py b/tests/unit/test_processing.py
@@ -220,7 +220,9 @@ def test_sklearn_with_all_parameters(
     sagemaker_session.process.assert_called_with(**expected_args)
 
 
-def test_local_mode_disables_local_code_by_default():
+@patch("sagemaker.config.config.load_sagemaker_config")
+def test_local_mode_disables_local_code_by_default(mock_load_config, base_config_with_schema):
+    mock_load_config.return_value = base_config_with_schema
     processor = Processor(
         image_uri="",
         role=ROLE,