diff --git a/src/sagemaker/tensorflow/estimator.py b/src/sagemaker/tensorflow/estimator.py index 83f3eb5f37..3f7df12777 100644 --- a/src/sagemaker/tensorflow/estimator.py +++ b/src/sagemaker/tensorflow/estimator.py @@ -302,7 +302,7 @@ def __init__( self._validate_args( py_version=py_version, script_mode=script_mode, - framework_version=framework_version, + framework_version=self.framework_version, training_steps=training_steps, evaluation_steps=evaluation_steps, requirements_file=requirements_file,