diff --git a/tests/tensorflow2/utils.py b/tests/tensorflow2/utils.py index 7b450db75..124da16b8 100644 --- a/tests/tensorflow2/utils.py +++ b/tests/tensorflow2/utils.py @@ -1,8 +1,7 @@ # Standard Library -from re import search - # Third Party import tensorflow.compat.v2 as tf +from packaging import version def is_tf_2_2(): @@ -13,6 +12,6 @@ def is_tf_2_2(): number of tensor_names emitted by 1. :return: bool """ - if search("2.2..", tf.__version__): + if version.parse(tf.__version__) >= version.parse("2.2.0"): return True return False