From 5c1b72f02952a1124cf2b9f8a2df3e9502688758 Mon Sep 17 00:00:00 2001 From: Peter Sobot Date: Thu, 8 Aug 2024 11:30:31 -0400 Subject: [PATCH] Update test_debugging.py --- tests/callbacks/test_debugging.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/callbacks/test_debugging.py b/tests/callbacks/test_debugging.py index aa389a6..867b0c5 100644 --- a/tests/callbacks/test_debugging.py +++ b/tests/callbacks/test_debugging.py @@ -38,7 +38,7 @@ def generate_data() -> Iterator[Tuple[List[int], List[int]]]: ), ) - model = tf.keras.Sequential([tf.keras.Input(shape=(1)), tf.keras.layers.Dense(1)]) + model = tf.keras.Sequential([tf.keras.Input(shape=(1,)), tf.keras.layers.Dense(1)]) model.compile(loss="binary_crossentropy") model.fit(fake_data, steps_per_epoch=1, epochs=10, callbacks=[HangDebugCallback()]) @@ -70,7 +70,7 @@ def generate_data() -> Iterator[Tuple[List[int], List[int]]]: ), ) - model = tf.keras.Sequential([tf.keras.Input(shape=(1)), tf.keras.layers.Dense(1)]) + model = tf.keras.Sequential([tf.keras.Input(shape=(1,)), tf.keras.layers.Dense(1)]) model.compile(loss="binary_crossentropy") model.fit( fake_data, @@ -114,7 +114,7 @@ def generate_data() -> Iterator[Tuple[List[int], List[int]]]: ), ) - model = tf.keras.Sequential([tf.keras.Input(shape=(1)), tf.keras.layers.Dense(1)]) + model = tf.keras.Sequential([tf.keras.Input(shape=(1,)), tf.keras.layers.Dense(1)]) model.compile(loss="binary_crossentropy") model.fit( fake_data,