address PR comment

Author: laurenyu

src/sagemaker/image_uris.py

@@ -101,9 +101,14 @@ def _validate_version_and_set_if_needed(version, config, framework):
     available_versions = list(config["versions"].keys())
 
     if len(available_versions) == 1:
-        logger.info(
-            "Defaulting to only available framework/algorithm version: %s", available_versions[0]
+        log_message = "Defaulting to the only supported framework/algorithm version: {}.".format(
+            available_versions[0]
         )
+        if version and version != available_versions[0]:
+            logger.warning("%s Ignoring framework/algorithm version: %s.", log_message, version)
+        elif not version:
+            logger.info(log_message)
+
         return available_versions[0]
 
     available_versions += list(config.get("version_aliases", {}).keys())

tests/unit/sagemaker/image_uris/test_retrieve.py

@@ -129,7 +129,7 @@ def test_retrieve_aliased_version(config_for_framework):
 
 
 @patch("sagemaker.image_uris.config_for_framework")
-def test_retrieve_default_version_if_possible(config_for_framework):
+def test_retrieve_default_version_if_possible(config_for_framework, caplog):
     config = copy.deepcopy(BASE_CONFIG)
     del config["versions"]["1.1.0"]
     config_for_framework.return_value = config
@@ -143,6 +143,17 @@ def test_retrieve_default_version_if_possible(config_for_framework):
     )
     assert "123412341234.dkr.ecr.us-west-2.amazonaws.com/dummy:1.0.0-cpu-py3" == uri
 
+    uri = image_uris.retrieve(
+        framework="useless-string",
+        version="invalid-version",
+        py_version="py3",
+        instance_type="ml.c4.xlarge",
+        region="us-west-2",
+        image_scope="training",
+    )
+    assert "123412341234.dkr.ecr.us-west-2.amazonaws.com/dummy:1.0.0-cpu-py3" == uri
+    assert "Ignoring framework/algorithm version: invalid-version." in caplog.text
+
 
 @patch("sagemaker.image_uris.config_for_framework", return_value=BASE_CONFIG)
 def test_retrieve_unsupported_version(config_for_framework):