feature: update TF framework version support (#2223)

* feature: update TF framework version support

* Update 1.15 inference

* Correct python versions

* Fix test parametrization

* Fix parametrization for major.minor versioning

Co-authored-by: Ajay Karpur <[email protected]>
Co-authored-by: Ahsan Khan <[email protected]>

Author: 3 people

src/sagemaker/fw_utils.py

@@ -59,7 +59,7 @@
     "local_gpu",
 )
 SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
-    "tensorflow": ["2.3.0", "2.3.1", "2.4.1"],
+    "tensorflow": ["2.3.1", "2.3.2", "2.4.1"],
     "pytorch": ["1.6.0", "1.7.1", "1.8.0"],
 }
 SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]