Skip to content

Commit c95a787

Browse files
committed
Make changes to prevent non 2.16 from making the change
1 parent e359e21 commit c95a787

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/tensorflow/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def __init__(
177177
fw.python_deprecation_warning(self._framework_name, defaults.LATEST_PY2_VERSION)
178178
)
179179
self.framework_version = framework_version
180-
self.override_fw_version = "2.16"
180+
self.override_fw_version = "2.16" if self.framework_version and "2.16" in self.framework_version else None
181181
# TF training and inference versions do not have a one-to-one connection. This mismatch
182182
# is accommodated by the underlying dictionary. The key of dictionary relates to the inference
183183
# version and the value relates to training version.

0 commit comments

Comments
 (0)