Cache TF Versions (awslabs#421)

NihalHarish · sophiayue1116 · commit 8c0b9d51bb31 · 2021-01-19T21:10:50.000-08:00
diff --git a/smdebug/tensorflow/utils.py b/smdebug/tensorflow/utils.py
@@ -428,7 +428,7 @@ def is_tf_version_greater_than_2_4_x():
 
 
 def is_tf_version_greater_than_2_4_x():
-    return version.parse("2.4.0") <= version.parse(tf.__version__)
+    return version.parse("2.4.0") <= TF_VERSION
 
 
 def is_profiler_supported_for_tf_version():
