From 6ec66fec74e5018c4faac186e5fd289955276608 Mon Sep 17 00:00:00 2001 From: Andrew Hundt Date: Tue, 10 Oct 2017 13:17:06 -0400 Subject: [PATCH] save_load_utils_test.py removes 'model.h5' --- tests/keras_contrib/utils/save_load_utils_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/keras_contrib/utils/save_load_utils_test.py b/tests/keras_contrib/utils/save_load_utils_test.py index 67f55fc..c2f840e 100644 --- a/tests/keras_contrib/utils/save_load_utils_test.py +++ b/tests/keras_contrib/utils/save_load_utils_test.py @@ -1,4 +1,5 @@ import pytest +import os from keras import backend as K from keras.layers import Input, Dense from keras.models import Model @@ -33,15 +34,16 @@ def test_save_and_load_all_weights(): ow1value[0, 0:3] = [4, 2, 0] K.set_value(ow1, ow1value) # save all weights - save_all_weights(m1, "model.h5") + save_all_weights(m1, 'model.h5') # new model m2 = make_model() # load all weights - load_all_weights(m2, "model.h5") + load_all_weights(m2, 'model.h5') # check weights assert_allclose(K.get_value(m2.layers[1].kernel)[0, 0:4], [1, 3, 3, 7]) # check optimizer weights assert_allclose(K.get_value(m2.optimizer.weights[3])[0, 0:3], [4, 2, 0]) + os.remove('model.h5') if __name__ == '__main__':