Skip to content

Commit 6056bae

Browse files
committed
fix: black-format and flake8 failures
1 parent fab744a commit 6056bae

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from __future__ import absolute_import
1414

1515
import pytest
16-
from packaging import version
16+
from packaging.version import parse
1717

1818
from sagemaker.huggingface import get_huggingface_llm_image_uri
1919
from 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

0 commit comments

Comments
 (0)