Skip to content

Commit 5502167

Browse files
author
Kim
committed
Add test for a warning with '0.90' since there is '0.90-2' available
1 parent 20fc81a commit 5502167

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/unit/test_amazon_estimator.py

+4
Original file line numberDiff line numberDiff line change
@@ -495,6 +495,10 @@ def test_get_xgboost_image_uri_warning_with_legacy(caplog):
495495
get_image_uri(REGION, "xgboost", 1)
496496
assert "There is a more up to date SageMaker XGBoost image." in caplog.text
497497

498+
def test_get_xgboost_image_uri_warning_with_no_sagemaker_version(caplog):
499+
get_image_uri(REGION, "xgboost", "0.90")
500+
assert "There is a more up to date SageMaker XGBoost image." in caplog.text
501+
498502

499503
def test_get_xgboost_image_uri_no_warning_with_latest(caplog):
500504
get_image_uri(REGION, "xgboost", XGBOOST_LATEST_VERSION.split("-")[0])

0 commit comments

Comments
 (0)