Use hasattr

Author: shantanutrip

src/sagemaker/estimator.py

@@ -3547,7 +3547,6 @@ def __init__(
         self.checkpoint_s3_uri = checkpoint_s3_uri
         self.checkpoint_local_path = checkpoint_local_path
         self.enable_sagemaker_metrics = enable_sagemaker_metrics
-        self.override_fw_version = None
 
     def _prepare_for_training(self, job_name=None):
         """Set hyperparameters needed for training. This method will also validate ``source_dir``.
@@ -3734,7 +3733,7 @@ def training_image_uri(self, region=None):
         return image_uris.get_training_image_uri(
             region=region or self.sagemaker_session.boto_region_name,
             framework=self._framework_name,
-            framework_version=self.override_fw_version or self.framework_version,  # pylint: disable=no-member
+            framework_version=self.override_fw_version if hasattr(self, "override_fw_version") else self.framework_version,  # pylint: disable=no-member
             py_version=self.py_version,  # pylint: disable=no-member
             image_uri=self.image_uri,
             distribution=getattr(self, "distribution", None),