Fix

Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
NVIDIA · Oct 24, 2024 · a3d2a55 · a3d2a55
1 parent e4a536e
commit a3d2a55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dali_tf_plugin/dali_tf_plugin_install_tool.py b/dali_tf_plugin/dali_tf_plugin_install_tool.py
@@ -366,7 +366,7 @@ def build(self):
             lib_path = os.path.join(self.plugin_dest_dir, lib_filename)
 
             # for a newer TF we need to compiler with C++17
-            cpp_ver = "--std=c++14" if self.tf_version < Version("2.10") else "--std=c++17"
+            cpp_ver = "--std=c++14" if Version(self.tf_version) < Version("2.10") else "--std=c++17"
             # Note: DNDEBUG flag is needed due to issue with TensorFlow custom ops:
             # https://github.com/tensorflow/tensorflow/issues/17316
             # Do not remove it.