Skip to content

Commit 582d2d5

Browse files
committed
Attempt a hack to handle missing 2.9.0 training image in tests
1 parent d53f5d5 commit 582d2d5

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/conftest.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -408,12 +408,16 @@ def tf_full_version(tensorflow_training_latest_version, tensorflow_inference_lat
408408
Fixture exists as such, since TF training and TFS have different latest versions.
409409
Otherwise, this would simply be a single latest version.
410410
"""
411-
return str(
411+
version = str(
412412
min(
413413
Version(tensorflow_training_latest_version),
414414
Version(tensorflow_inference_latest_version),
415415
)
416416
)
417+
# Hack. See https://github.com/aws/sagemaker-python-sdk/pull/3251
418+
if version === "2.9.0":
419+
return "2.9"
420+
return version
417421

418422

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

0 commit comments

Comments
 (0)