Skip to content

Commit 9f0b0f0

Browse files
committed
fixed keras tests
1 parent 22c97e0 commit 9f0b0f0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/batch_generators/test_batch_generator_keras.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def test_fit_multiple_epochs(self):
7171
keras_model.fit(keras_bg, epochs=2)
7272

7373
def test_multiple_inputs(self):
74-
keras_bg = BatchGenerator(self.data, x_structure=[('x1', None), ('x2', None)],
74+
keras_bg = BatchGenerator(self.data, x_structure=(('x1', None), ('x2', None)),
7575
y_structure=('y1', self.y1_enc))
7676
inp1 = keras.layers.Input(shape=(1,))
7777
inp2 = keras.layers.Input(shape=(1,))
@@ -82,7 +82,7 @@ def test_multiple_inputs(self):
8282
keras_model.fit(keras_bg, epochs=2)
8383

8484
def test_multiple_inputs_outputs(self):
85-
keras_bg = BatchGenerator(self.data, x_structure=[('x1', None), ('x2', None)],
85+
keras_bg = BatchGenerator(self.data, x_structure=(('x1', None), ('x2', None)),
8686
y_structure=[('y1', self.y1_enc), ('y2', self.y2_enc)])
8787
inp1 = keras.layers.Input(shape=(1,))
8888
inp2 = keras.layers.Input(shape=(1,))

0 commit comments

Comments
 (0)