Test with making changes to model.py

shantanutrip · shantanutrip · commit 2a2dfd05a626 · 2024-08-14T20:36:02.000Z
diff --git a/src/sagemaker/tensorflow/model.py b/src/sagemaker/tensorflow/model.py
@@ -199,7 +199,7 @@ def __init__(
         # patch versions, but end up hosting the model of same TF version. For eg., the upstream
         # TFS-2.12.0 release was a bad release and hence a new TFS-2.12.1 release was made to host
         # models from TF-2.12.0.
-        training_inference_version_mismatch_dict = {"2.12.0": "2.12.1"}
+        training_inference_version_mismatch_dict = {"2.12.0": "2.12.1", "2.16.2": "2.16.1"}
         self.inference_framework_version = training_inference_version_mismatch_dict.get(
             framework_version, framework_version
         )

Original file line number	Diff line number	Diff line change
`@@ -199,7 +199,7 @@ def __init__(`
`199`	`199`	`# patch versions, but end up hosting the model of same TF version. For eg., the upstream`
`200`	`200`	`# TFS-2.12.0 release was a bad release and hence a new TFS-2.12.1 release was made to host`
`201`	`201`	`# models from TF-2.12.0.`
`202`		`- training_inference_version_mismatch_dict = {"2.12.0": "2.12.1"}`
	`202`	`+ training_inference_version_mismatch_dict = {"2.12.0": "2.12.1", "2.16.2": "2.16.1"}`
`203`	`203`	`self.inference_framework_version = training_inference_version_mismatch_dict.get(`
`204`	`204`	`framework_version, framework_version`
`205`	`205`	`)`