add TF 2.4.0 to fw_utils and test

Author: Ubuntu

src/sagemaker/fw_utils.py

@@ -57,7 +57,7 @@
     "local_gpu",
 )
 SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
-    "tensorflow": ["2.3.0", "2.3.1"],
+    "tensorflow": ["2.3.0", "2.3.1", "2.4.0"],
     "pytorch": ["1.6.0"],
 }
 SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]

tests/unit/test_fw_utils.py

@@ -631,6 +631,7 @@ def test_validate_smdataparallel_args_not_raises():
     good_args = [
         (None, None, None, None, smdataparallel_disabled),
         ("ml.p3.16xlarge", "tensorflow", "2.3.1", "py3", smdataparallel_enabled),
+        ("ml.p3.16xlarge", "tensorflow", "2.4.0", "py3", smdataparallel_enabled),
         ("ml.p3.16xlarge", "pytorch", "1.6.0", "py3", smdataparallel_enabled),
     ]
     for instance_type, framework_name, framework_version, py_version, distribution in good_args: