use predict so test works for non-egaer execution
diff --git a/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py b/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py
index 0202ceb..d3f25a0 100644
--- a/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py
+++ b/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py
@@ -476,7 +476,11 @@
standardize = "".join(["lower", "_and_strip_punctuation"])
layer = get_layer_class()(standardize=standardize)
- output = layer(input_array).numpy()
+
+ input_data = keras.Input(shape=(1,), dtype=dtypes.string)
+ output_data = layer(input_data)
+ model = keras.Model(inputs=input_data, outputs=output_data)
+ output = model.predict(input_array)
self.assertAllEqual(expected_output, output)
@@ -486,7 +490,11 @@
split = "".join(["white", "space"])
layer = get_layer_class()(split=split)
- output = layer(input_array).numpy()
+
+ input_data = keras.Input(shape=(1,), dtype=dtypes.string)
+ output_data = layer(input_data)
+ model = keras.Model(inputs=input_data, outputs=output_data)
+ output = model.predict(input_array)
self.assertAllEqual(expected_output, output)
