Small change to expected directory structure in golden_analyzer_cache…

…_test and output_wrapper_test PiperOrigin-RevId: 372386921
tensorflow · May 6, 2021 · 774458b · 774458b
1 parent df21282
commit 774458b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tensorflow_transform/beam/analyzer_cache.py b/tensorflow_transform/beam/analyzer_cache.py
@@ -35,8 +35,9 @@
 # This should be advanced whenever a non-backwards compatible change is made
 # that affects analyzer cache. For example, changing accumulator format.
 _CACHE_VERSION_NUMBER = 1
-_CACHE_VERSION = tf.compat.as_bytes('__v{}__{}.{}_'.format(
-    _CACHE_VERSION_NUMBER, sys.version_info.major, sys.version_info.minor))
+_PYTHON_VERSION = '{}.{}'.format(sys.version_info.major, sys.version_info.minor)
+_CACHE_VERSION = tf.compat.as_bytes('__v{}__{}_'.format(_CACHE_VERSION_NUMBER,
+                                                        _PYTHON_VERSION))
 
 _CACHE_COMPONENT_CHARACTER_REPLACEMENTS = (
     ('/', '-'),