Skip to content

Commit 22c97e0

Browse files
committed
fixed keras tests
1 parent af64ebc commit 22c97e0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/batch_generators/test_batch_generator_tf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def test_fit_multiple_epochs(self):
7676
tf_model.fit(tf_bg, epochs=2)
7777

7878
def test_multiple_inputs(self):
79-
keras_bg = BatchGeneratorTF(self.data, x_structure=[('x1', None), ('x2', None)],
79+
keras_bg = BatchGeneratorTF(self.data, x_structure=(('x1', None), ('x2', None)),
8080
y_structure=('y1', self.y1_enc))
8181
inp1 = tf.keras.layers.Input(shape=(1,))
8282
inp2 = tf.keras.layers.Input(shape=(1,))
@@ -87,7 +87,7 @@ def test_multiple_inputs(self):
8787
keras_model.fit(keras_bg, epochs=2)
8888

8989
def test_multiple_inputs_outputs(self):
90-
keras_bg = BatchGeneratorTF(self.data, x_structure=[('x1', None), ('x2', None)],
90+
keras_bg = BatchGeneratorTF(self.data, x_structure=(('x1', None), ('x2', None)),
9191
y_structure=[('y1', self.y1_enc), ('y2', self.y2_enc)])
9292
inp1 = tf.keras.layers.Input(shape=(1,))
9393
inp2 = tf.keras.layers.Input(shape=(1,))

0 commit comments

Comments
 (0)