Skip to content

Commit f47e3f3

Browse files
committed
Attempt a hack to handle missing 2.9.0 training image in tests
1 parent 8fdeda4 commit f47e3f3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/conftest.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -384,12 +384,16 @@ def tf_full_version(tensorflow_training_latest_version, tensorflow_inference_lat
384384
Fixture exists as such, since TF training and TFS have different latest versions.
385385
Otherwise, this would simply be a single latest version.
386386
"""
387-
return str(
387+
version = str(
388388
min(
389389
Version(tensorflow_training_latest_version),
390390
Version(tensorflow_inference_latest_version),
391391
)
392392
)
393+
# Hack. See https://github.com/aws/sagemaker-python-sdk/pull/3251
394+
if version === "2.9.0":
395+
return "2.9"
396+
return version
393397

394398

395399
@pytest.fixture(scope="module")

0 commit comments

Comments
 (0)