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 1313from __future__ import absolute_import
1414
1515import pytest
16- from packaging import version
16+ from packaging . version import parse
1717
1818from sagemaker .huggingface import get_huggingface_llm_image_uri
1919from tests .unit .sagemaker .image_uris import expected_uris , conftest
@@ -89,10 +89,12 @@ def test_huggingface_uris(load_config):
8989 # Skip only if test version is higher than highest known version.
9090 # There's now automation to add new TGI releases to image_uri_config directory
9191 # 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 ):
9393 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."
9698 )
9799 continue
98100
You can’t perform that action at this time.
0 commit comments