Fix test options for TF 2.8 for fw_utils

Author: saimidu

tests/unit/test_fw_utils.py

@@ -686,7 +686,7 @@ def test_validate_smdataparallel_args_not_raises():
         ("ml.p3.16xlarge", "tensorflow", "2.6.0", "py38", smdataparallel_enabled),
         ("ml.p3.16xlarge", "tensorflow", "2.6.2", "py38", smdataparallel_enabled),
         ("ml.p3.16xlarge", "tensorflow", "2.6", "py38", smdataparallel_enabled),
-        ("ml.p3.16xlarge", "tensorflow", "2.8.0", "py38", smdataparallel_enabled),
+        ("ml.p3.16xlarge", "tensorflow", "2.8.0", "py39", smdataparallel_enabled),
         ("ml.p3.16xlarge", "tensorflow", "2.8", "py39", smdataparallel_enabled),
         ("ml.p3.16xlarge", "pytorch", "1.6.0", "py3", smdataparallel_enabled),
         ("ml.p3.16xlarge", "pytorch", "1.6", "py3", smdataparallel_enabled),
@@ -702,7 +702,7 @@ def test_validate_smdataparallel_args_not_raises():
         ("ml.p3.16xlarge", "tensorflow", "2.4.3", "py37", smdataparallel_enabled_custom_mpi),
         ("ml.p3.16xlarge", "tensorflow", "2.5.1", "py37", smdataparallel_enabled_custom_mpi),
         ("ml.p3.16xlarge", "tensorflow", "2.6.2", "py38", smdataparallel_enabled_custom_mpi),
-        ("ml.p3.16xlarge", "tensorflow", "2.8.0", "py38", smdataparallel_enabled_custom_mpi),
+        ("ml.p3.16xlarge", "tensorflow", "2.8.0", "py39", smdataparallel_enabled_custom_mpi),
         ("ml.p3.16xlarge", "pytorch", "1.8.0", "py3", smdataparallel_enabled_custom_mpi),
         ("ml.p3.16xlarge", "pytorch", "1.9.1", "py38", smdataparallel_enabled_custom_mpi),
     ]