Fix dependency manager ut

samruds · samruds · commit 10a5b8fb8813 · 2024-03-29T15:45:50.000-07:00
diff --git a/tests/unit/sagemaker/serve/detector/test_dependency_manager.py b/tests/unit/sagemaker/serve/detector/test_dependency_manager.py
@@ -99,7 +99,7 @@ def test_capture_dependencies(self, mock_subprocess, mock_file, mock_path):
             call("custom_module==1.2.3\n"),
             call("numpy==4.5\n"),
             call("boto3=1.28.*\n"),
-            call("sagemaker>=2.199\n"),
+            call("sagemaker[huggingface]>=2.199\n"),
             call("other_module@http://some/website.whl\n"),
         ]
         mocked_writes.assert_has_calls(expected_calls)

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,7 @@ def test_capture_dependencies(self, mock_subprocess, mock_file, mock_path):`
`99`	`99`	`call("custom_module==1.2.3\n"),`
`100`	`100`	`call("numpy==4.5\n"),`
`101`	`101`	`call("boto3=1.28.*\n"),`
`102`		`- call("sagemaker>=2.199\n"),`
	`102`	`+ call("sagemaker[huggingface]>=2.199\n"),`
`103`	`103`	`call("other_module@http://some/website.whl\n"),`
`104`	`104`	`]`
`105`	`105`	`mocked_writes.assert_has_calls(expected_calls)`