File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
tests/unit/sagemaker/image_uris Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change 13
13
from __future__ import absolute_import
14
14
15
15
import pytest
16
- from packaging import version
16
+ from packaging . version import parse
17
17
18
18
from sagemaker .huggingface import get_huggingface_llm_image_uri
19
19
from tests .unit .sagemaker .image_uris import expected_uris , conftest
@@ -89,10 +89,12 @@ def test_huggingface_uris(load_config):
89
89
# Skip only if test version is higher than highest known version.
90
90
# There's now automation to add new TGI releases to image_uri_config directory
91
91
# that doesn't involve a human raising a PR.
92
- if version . parse (version ) > version . parse (highest_version ):
92
+ if parse (version ) > parse (highest_version ):
93
93
print (
94
- f"Skipping test for version { test_version } as it is higher than the highest known version { highest_version } . "
95
- "There is automation that now updates the image_uri_config without a human raising a PR."
94
+ f"Skipping test for version { version } as it is higher than "
95
+ "the highest known version {highest_version}. There is "
96
+ "automation that now updates the image_uri_config "
97
+ "without a human raising a PR."
96
98
)
97
99
continue
98
100
You can’t perform that action at this time.
0 commit comments