We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e359e21 commit c95a787Copy full SHA for c95a787
src/sagemaker/tensorflow/estimator.py
@@ -177,7 +177,7 @@ def __init__(
177
fw.python_deprecation_warning(self._framework_name, defaults.LATEST_PY2_VERSION)
178
)
179
self.framework_version = framework_version
180
- self.override_fw_version = "2.16"
+ self.override_fw_version = "2.16" if self.framework_version and "2.16" in self.framework_version else None
181
# TF training and inference versions do not have a one-to-one connection. This mismatch
182
# is accommodated by the underlying dictionary. The key of dictionary relates to the inference
183
# version and the value relates to training version.
0 commit comments