Update Travis config to test on TF 2
@@ -13,9 +13,13 @@ matrix:
- python: 3.6
env: KERAS_BACKEND=tensorflow TEST_MODE=API
- python: 2.7
- env: KERAS_BACKEND=tensorflow
+ env: KERAS_BACKEND=tensorflow-1
- python: 3.6
- env: KERAS_BACKEND=tensorflow
+ env: KERAS_BACKEND=tensorflow-1
+ - python: 2.7
+ env: KERAS_BACKEND=tensorflow-2
+ - python: 3.6
+ env: KERAS_BACKEND=tensorflow-2
- python: 2.7
env: KERAS_BACKEND=theano THEANO_FLAGS=optimizer=fast_compile MKL="mkl mkl-service" RUN_ONLY_BACKEND_TESTS=1
- python: 3.6
@@ -56,7 +60,11 @@ install:
- pip install -e .[tests] --progress-bar off
# install TensorFlow (CPU version).
- - pip install tensorflow==1.13.1 --progress-bar off
+ - if [[ "$KERAS_BACKEND" == "tensorflow-2" ]]; then
+ pip install tf-nightly-2.0-preview --progress-bar off
+ else
+ pip install tensorflow==1.13.1 --progress-bar off
+ fi
# install cntk
- if [[ "$KERAS_BACKEND" == "cntk" ]] || [[ "$TEST_MODE" == "PEP8_DOC" ]] || [[ "$TEST_MODE" == "API" ]]; then
