Skip to content

Commit 31e22ff

Browse files
author
Bryannah Hernandez
committed
test fix
1 parent 8dd3468 commit 31e22ff

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/unit/sagemaker/serve/builder/test_requirements_manager.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class TestRequirementsManager(unittest.TestCase):
2929
@patch(
3030
"sagemaker.serve.builder.requirements_manager.RequirementsManager._detect_conda_env_and_local_dependencies"
3131
)
32-
def test_capture_and_install_dependencies(
32+
def test_capture_and_install_dependencies_txt(
3333
self,
3434
mock_detect_conda_env_and_local_dependencies,
3535
mock_install_requirements_txt,
@@ -40,8 +40,7 @@ def test_capture_and_install_dependencies(
4040
RequirementsManager().capture_and_install_dependencies()
4141
mock_install_requirements_txt.assert_called_once()
4242

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")
4544
mock_update_conda_env_in_path.assert_called_once()
4645

4746
@patch(

0 commit comments

Comments
 (0)