Skip to content

Commit 5711a20

Browse files
committed
Fix print and use getattr
1 parent 3036cce commit 5711a20

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3728,12 +3728,12 @@ def training_image_uri(self, region=None):
37283728
str: The URI of the Docker image.
37293729
"""
37303730
## TODO: Revert
3731-
print(f"ABCD1234 self.override_fw_version: {self.override_fw_version} and self.framework_version: {self.framework_version}")
3731+
print(f"ABCD1234 self.override_fw_version: {getattr(self, "override_fw_version", None)} and self.framework_version: {self.framework_version}")
37323732

37333733
return image_uris.get_training_image_uri(
37343734
region=region or self.sagemaker_session.boto_region_name,
37353735
framework=self._framework_name,
3736-
framework_version=self.override_fw_version if hasattr(self, "override_fw_version") else self.framework_version, # pylint: disable=no-member
3736+
framework_version= getattr(self, "override_fw_version", self.framework_version), # pylint: disable=no-member
37373737
py_version=self.py_version, # pylint: disable=no-member
37383738
image_uri=self.image_uri,
37393739
distribution=getattr(self, "distribution", None),

0 commit comments

Comments
 (0)