diff --git a/tensorflow/lite/experimental/microfrontend/python/kernel_tests/audio_microfrontend_op_test.py b/tensorflow/lite/experimental/microfrontend/python/kernel_tests/audio_microfrontend_op_test.py index 8a9f30beafd..913c330ef0e 100644 --- a/tensorflow/lite/experimental/microfrontend/python/kernel_tests/audio_microfrontend_op_test.py +++ b/tensorflow/lite/experimental/microfrontend/python/kernel_tests/audio_microfrontend_op_test.py @@ -22,7 +22,6 @@ import tensorflow as tf from tensorflow.lite.experimental.microfrontend.python.ops import audio_microfrontend_op as frontend_op from tensorflow.python.framework import ops -from tensorflow.python.framework import test_util SAMPLE_RATE = 1000 WINDOW_SIZE = 25 @@ -57,7 +56,6 @@ class AudioFeatureGenerationTest(tf.test.TestCase): self.assertAllEqual(filterbanks.eval(), [[479, 425], [436, 378], [410, 350], [391, 325]]) - @test_util.run_v1_only("b/120545219") def testSimpleFloatScaled(self): with self.test_session(): audio = tf.constant( @@ -79,7 +77,6 @@ class AudioFeatureGenerationTest(tf.test.TestCase): [[7.484375, 6.640625], [6.8125, 5.90625], [6.40625, 5.46875], [6.109375, 5.078125]]) - @test_util.run_v1_only("b/120545219") def testStacking(self): with self.test_session(): audio = tf.constant( @@ -122,7 +119,6 @@ class AudioFeatureGenerationTest(tf.test.TestCase): [[479, 425, 479, 425, 436, 378], [479, 425, 436, 378, 410, 350], [436, 378, 410, 350, 391, 325], [410, 350, 391, 325, 391, 325]]) - @test_util.run_v1_only("b/120545219") def testStackingDropFrame(self): with self.test_session(): audio = tf.constant(