Bump up TensorFlow version in tests

- the latest to 2.15 - previous to 2.14.1 Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
NVIDIA · Nov 23, 2023 · badf7e9 · badf7e9
1 parent 2e873d7
commit badf7e9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/dali/python/nvidia/dali/plugin/tf.py b/dali/python/nvidia/dali/plugin/tf.py
@@ -398,7 +398,6 @@ def _get_current_device_spec():
 
 
 if dataset_compatible_tensorflow():
-    from tensorflow.python.framework import ops
     from tensorflow.python.data.ops import dataset_ops
     from tensorflow.python.data.util import structure
     import functools
@@ -451,7 +450,7 @@ def __init__(
                 output_dtypes = (output_dtypes, )
                 output_shapes = (output_shapes, )
 
-            output_classes = nest.map_structure(lambda _: ops.Tensor, output_dtypes)
+            output_classes = nest.map_structure(lambda _: tf.Tensor, output_dtypes)
 
             self._pipeline_instance = pipeline  # keep the live Pipeline object
             self._pipeline_serialized = serialize_pipeline(pipeline)

diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -470,10 +470,14 @@ def get_pyvers_name(self, url, cuda_version):
                             "cupy-cuda11x"),
                 CudaPackage("tensorflow-gpu",
                             {"110": [
-                                PckgVer("2.12.1", python_min_ver="3.8", alias="tensorflow",
+                                PckgVer("2.14.1", python_min_ver="3.8", alias="tensorflow",
+                                        dependencies=["protobuf<4", "numpy<1.24",
+                                                      "urllib3<2.0", "typing_extensions<4.6"])],
+                             "120": [
+                                PckgVer("2.14.1", python_min_ver="3.8", alias="tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24",
                                                       "urllib3<2.0", "typing_extensions<4.6"]),
-                                PckgVer("2.13.0", python_min_ver="3.8", alias="tensorflow",
+                                PckgVer("2.15.0", python_min_ver="3.8", alias="tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24",
                                                       "urllib3<2.0", "typing_extensions<4.6"])]}),
                 CudaPackageExtraIndex("torch",