fix: pin tf version in tests

Author: benieric

tests/data/tensorflow_mnist/mnist_v2.py

@@ -198,7 +198,7 @@ def main(args):
 
         if args.current_host == args.hosts[0]:
             ckpt_manager.save()
-            net.save("/opt/ml/model/1.keras")
+            net.save("/opt/ml/model/1")
 
 
 if __name__ == "__main__":

tests/integ/test_tf.py

@@ -278,7 +278,7 @@ def _create_and_fit_estimator(sagemaker_session, tf_version, py_version, instanc
         instance_count=2,
         instance_type=instance_type,
         sagemaker_session=sagemaker_session,
-        framework_version=tf_version,
+        framework_version="2.14",
         py_version=py_version,
         distribution=PARAMETER_SERVER_DISTRIBUTION,
         disable_profiler=True,

tests/integ/test_tuner.py

@@ -701,7 +701,7 @@ def test_tuning_tf(
         instance_count=1,
         instance_type=cpu_instance_type,
         sagemaker_session=sagemaker_session,
-        framework_version=tensorflow_training_latest_version,
+        framework_version="2.14",
         py_version=tensorflow_training_latest_py_version,
     )
 
@@ -749,7 +749,7 @@ def test_tuning_tf_vpc_multi(
         entry_point=script_path,
         source_dir=resource_path,
         role="SageMakerRole",
-        framework_version=tensorflow_training_latest_version,
+        framework_version="2.14",
         py_version=tensorflow_training_latest_py_version,
         instance_count=instance_count,
         instance_type=instance_type,