File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
tests/unit/sagemaker/serve/builder Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ class TestRequirementsManager(unittest.TestCase):
29
29
@patch (
30
30
"sagemaker.serve.builder.requirements_manager.RequirementsManager._detect_conda_env_and_local_dependencies"
31
31
)
32
- def test_capture_and_install_dependencies (
32
+ def test_capture_and_install_dependencies_txt (
33
33
self ,
34
34
mock_detect_conda_env_and_local_dependencies ,
35
35
mock_install_requirements_txt ,
@@ -40,8 +40,7 @@ def test_capture_and_install_dependencies(
40
40
RequirementsManager ().capture_and_install_dependencies ()
41
41
mock_install_requirements_txt .assert_called_once ()
42
42
43
- mock_detect_conda_env_and_local_dependencies .side_effect = lambda : ".yml"
44
- RequirementsManager ().capture_and_install_dependencies ()
43
+ RequirementsManager ().capture_and_install_dependencies ("conda.yml" )
45
44
mock_update_conda_env_in_path .assert_called_once ()
46
45
47
46
@patch (
You can’t perform that action at this time.
0 commit comments