rollback attribute changes which is not compatible with TF 1.15

tensorflow · Nov 27, 2024 · 99c5f2a · 99c5f2a
1 parent f2442ea
commit 99c5f2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tfx/examples/mnist/mnist_utils_native_keras_base.py b/tfx/examples/mnist/mnist_utils_native_keras_base.py
@@ -69,7 +69,7 @@ def build_keras_model() -> tf.keras.Model:
   model = tf.keras.Sequential()
   model.add(
       tf.keras.layers.InputLayer(
-          shape=(784,), name=transformed_name(IMAGE_KEY)))
+          input_shape=(784,), name=transformed_name(IMAGE_KEY)))
   model.add(tf.keras.layers.Dense(64, activation='relu'))
   model.add(tf.keras.layers.Dropout(0.2))
   model.add(tf.keras.layers.Dense(64, activation='relu'))