Removed usage of keras_parametrized. (#1271)

Author: gabrieldemarmiesse

tensorflow_addons/seq2seq/basic_decoder_test.py

@@ -24,11 +24,9 @@
 from tensorflow_addons.seq2seq import attention_wrapper
 from tensorflow_addons.seq2seq import basic_decoder
 from tensorflow_addons.seq2seq import sampler as sampler_py
-from tensorflow_addons.utils import test_utils
 
 
-@test_utils.keras_parameterized.run_all_keras_modes
-class BasicDecoderTest(test_utils.keras_parameterized.TestCase):
+class BasicDecoderTest(tf.test.TestCase, parameterized.TestCase):
     """Unit test for basic_decoder.BasicDecoder."""
 
     @parameterized.named_parameters(

tensorflow_addons/seq2seq/decoder_test.py

@@ -20,13 +20,11 @@
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_addons.utils import test_utils
 from tensorflow_addons.seq2seq import basic_decoder
 from tensorflow_addons.seq2seq import sampler as sampler_py
 
 
-@test_utils.keras_parameterized.run_all_keras_modes
-class DecodeRNNTest(test_utils.keras_parameterized.TestCase, tf.test.TestCase):
+class DecodeRNNTest(tf.test.TestCase):
     """Tests for Decoder."""
 
     def _testDecodeRNN(self, time_major, maximum_iterations=None):

tensorflow_addons/utils/test_utils.py

@@ -30,7 +30,6 @@
     run_in_graph_and_eager_modes,
 )
 from tensorflow.python.keras.testing_utils import layer_test  # noqa: F401
-from tensorflow.python.keras import keras_parameterized  # noqa: F401
 
 
 @contextlib.contextmanager