From 70e68c11f569ce2af145054ebeca09efa8ca6503 Mon Sep 17 00:00:00 2001 From: Andrew Hundt Date: Sun, 15 Oct 2017 13:28:43 -0400 Subject: [PATCH] save_load_utils_test.py make tensorflow only Making this test tensorflow only until https://github.com/farizrahman4u/keras-contrib/issues/157 is resolved. --- tests/keras_contrib/utils/save_load_utils_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/keras_contrib/utils/save_load_utils_test.py b/tests/keras_contrib/utils/save_load_utils_test.py index c2f840e..a11e826 100644 --- a/tests/keras_contrib/utils/save_load_utils_test.py +++ b/tests/keras_contrib/utils/save_load_utils_test.py @@ -4,10 +4,13 @@ from keras import backend as K from keras.layers import Input, Dense from keras.models import Model from numpy.testing import assert_allclose +from keras.utils.test_utils import keras_test from keras_contrib.utils.save_load_utils import save_all_weights, load_all_weights +@pytest.mark.skipif(K.backend() != 'tensorflow', reason='save_all_weights and load_all_weights only supported on TensorFlow') +@keras_test def test_save_and_load_all_weights(): ''' Test save_all_weights and load_all_weights. Save and load optimizer and model weights but not configuration.