added test
This commit is contained in:
parent
17b7e16913
commit
57eccc7bc2
@ -3328,6 +3328,16 @@ class TestTrainingWithMetrics(keras_parameterized.TestCase):
|
||||
outer_model.fit(np.ones((10, 1)), np.ones((10, 1)), batch_size=10)
|
||||
self.assertEqual([m.name for m in outer_model.metrics],
|
||||
['loss', 'acc2', 'mean', 'mean1', 'mean2'])
|
||||
|
||||
def test_subclassed_model_with_empty_list_attr(self):
|
||||
class ModelSubclass(training_module.Model):
|
||||
def __init__(self):
|
||||
self.empty_list = []
|
||||
inputs = layers_module.Input(shape=())
|
||||
outputs = inputs + 1
|
||||
super(ModelSubclass, self).__init__(inputs, outputs)
|
||||
|
||||
ModelSubclass() # empty_list attr assignment should not raise
|
||||
|
||||
|
||||
class BareUpdateLayer(layers_module.Layer):
|
||||
|
Loading…
Reference in New Issue
Block a user