update error message

Author: laurenyu

src/sagemaker/tensorflow/estimator.py

@@ -161,10 +161,6 @@ def _validate_args(self, py_version, framework_version):
             raise AttributeError(msg)
 
         if self._only_legacy_mode_supported() and self.image_name is None:
-            additional_instructions = ""
-            if self.model_dir is not False:
-                additional_instructions = " and set 'model_dir=False'"
-
             legacy_image_uri = fw.create_image_uri(
                 self.sagemaker_session.boto_region_name,
                 "tensorflow",
@@ -176,10 +172,10 @@ def _validate_args(self, py_version, framework_version):
             # TODO: add link to docs to explain how to use legacy mode with v2
             msg = (
                 "TF {} supports only legacy mode. Please supply the image URI directly with "
-                "'image_name={}'{}. If you were using any legacy mode parameters "
+                "'image_name={}' and set 'model_dir=False'. If you are using any legacy parameters "
                 "(training_steps, evaluation_steps, checkpoint_path, requirements_file), "
                 "make sure to pass them directly as hyperparameters instead."
-            ).format(self.framework_version, legacy_image_uri, additional_instructions)
+            ).format(self.framework_version, legacy_image_uri)
 
             raise ValueError(msg)
 

tests/unit/sagemaker/tensorflow/test_estimator_init.py

@@ -118,7 +118,7 @@ def test_require_image_name_if_fw_ver_is_less_than_1_11(sagemaker_session, tf_ve
         "TF {version} supports only legacy mode. Please supply the image URI directly with "
         "'image_name=520713654638.dkr.ecr.{region}.amazonaws.com/"
         "sagemaker-tensorflow:{version}-cpu-py2' and set 'model_dir=False'. "
-        "If you were using any legacy mode parameters (training_steps, evaluation_steps, "
+        "If you are using any legacy parameters (training_steps, evaluation_steps, "
         "checkpoint_path, requirements_file), make sure to pass them directly as hyperparameters instead."
     ).format(version=tf_version, region=REGION)