Skip new cached ptransform test for TF1 when tf_compat_v1=False.

PiperOrigin-RevId: 421402300
tensorflow · Jan 12, 2022 · 0e614ca · 0e614ca
1 parent f95c760
commit 0e614ca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow_transform/beam/cached_impl_test.py b/tensorflow_transform/beam/cached_impl_test.py
@@ -1099,6 +1099,8 @@ def preprocessing_fn(inputs):
   @tft_unit.named_parameters(_TF_VERSION_NAMED_PARAMETERS)
   @mock_out_cache_hash
   def test_cached_ptransform_analyzer(self, use_tf_compat_v1):
+    if not use_tf_compat_v1:
+      tft_unit.skip_if_not_tf2('Tensorflow 2.x required.')
 
     class _AnalyzerMakeAccumulators(beam.PTransform):