Simplify pickling tests.
import pytest
-import os
import sys
-import tempfile
import numpy as np
from numpy.testing import assert_allclose
-from numpy.testing import assert_raises
-from keras import backend as K
from keras.models import Model, Sequential
from keras.layers import Dense, Lambda, RepeatVector, TimeDistributed
from keras.layers import Input
@@ -20,12 +16,6 @@ else:
import cPickle as pickle
-skipif_no_tf_gpu = pytest.mark.skipif(
- (K.backend() != 'tensorflow') or
- (not K.tensorflow_backend._get_available_gpus()),
- reason='Requires TensorFlow backend and a GPU')
-
-
def test_sequential_model_pickling():
model = Sequential()
model.add(Dense(2, input_shape=(3,)))
