Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Merge master
  • Loading branch information
Dref360 committed Nov 27, 2018
commit dc05b52da19407eca79f7a6b905ce7c1990a51e0
5 changes: 3 additions & 2 deletions tests/keras/utils/data_utils_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,10 @@
import six
from six.moves.urllib.parse import urljoin
from six.moves.urllib.request import pathname2url
import time
import warnings

from flaky import flaky

from keras.utils import GeneratorEnqueuer
from keras.utils import OrderedEnqueuer
from keras.utils import Sequence
Expand All @@ -28,7 +29,7 @@
six.PY2 and 'TRAVIS_PYTHON_VERSION' in os.environ,
reason='Temporarily disabled until the use_multiprocessing problem is solved')

skip_generators = pytest.mark.skipif(K.backend() in ['tensorflow', 'cntk'] and
skip_generators = pytest.mark.skipif(K.backend() in {'tensorflow', 'cntk'} and
'TRAVIS_PYTHON_VERSION' in os.environ,
reason='Generators do not work with `spawn`.')

Expand Down
6 changes: 3 additions & 3 deletions tests/test_multiprocessing.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
from keras.utils import Sequence
from keras import backend as K

pytestmark = pytest.mark.skipif('TRAVIS_PYTHON_VERSION' in os.environ,
reason='Temporarily disabled until'
'the use_multiprocessing problem is solved')
pytestmark = pytest.mark.skipif(
K.backend() in {'tensorflow', 'cntk'} and 'TRAVIS_PYTHON_VERSION' in os.environ,
reason='Temporarily disabled until the use_multiprocessing problem is solved')

STEPS_PER_EPOCH = 100
STEPS = 100
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.