We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 20fc81a commit 5502167Copy full SHA for 5502167
tests/unit/test_amazon_estimator.py
@@ -495,6 +495,10 @@ def test_get_xgboost_image_uri_warning_with_legacy(caplog):
495
get_image_uri(REGION, "xgboost", 1)
496
assert "There is a more up to date SageMaker XGBoost image." in caplog.text
497
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
+
502
503
def test_get_xgboost_image_uri_no_warning_with_latest(caplog):
504
get_image_uri(REGION, "xgboost", XGBOOST_LATEST_VERSION.split("-")[0])
0 commit comments