update

Author: Qingzi-Lan

src/sagemaker/model.py

@@ -627,7 +627,7 @@ def _compilation_job_config(
             input_model_config["FrameworkVersion"] = utils.get_short_version(framework_version)
 
         if framework.lower() == "tensorflow":
-            input_model_config["FrameworkVersion"] = framework_version
+            input_model_config["FrameworkVersion"] = utils.get_short_version(framework_version)
 
         role = self.sagemaker_session.expand_role(role)
         output_model_config = {