Skip to content

Commit e25e2d0

Browse files
committed
fixed keras tests
1 parent 9f0b0f0 commit e25e2d0

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

tests/batch_generators/test_batch_generator_keras.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def test_multiple_inputs(self):
8383

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

tests/batch_generators/test_batch_generator_tf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def test_multiple_inputs(self):
8888

8989
def test_multiple_inputs_outputs(self):
9090
keras_bg = BatchGeneratorTF(self.data, x_structure=(('x1', None), ('x2', None)),
91-
y_structure=[('y1', self.y1_enc), ('y2', self.y2_enc)])
91+
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,))
9494
x = tf.keras.layers.Concatenate()([inp1, inp2])

0 commit comments

Comments
 (0)