From 16e415d1fddb960706909b805cba4ac4e2a4cbed Mon Sep 17 00:00:00 2001 From: Marco Raveri Date: Sun, 22 Sep 2024 11:12:07 +0200 Subject: [PATCH] Update --- .github/workflows/test.yml | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 469cf02..503ef37 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -46,7 +46,14 @@ jobs: python -m pip install --upgrade pip pip install tensorflow==${{ matrix.tensorflow-version }} pip install tensorflow-probability==${{ matrix.tfp-version }} - pip install tf-keras + # Install tf-keras for tensorflow 2.16.1 + if [ "${{ matrix.tensorflow-version }}" == "2.16.1" ]; then + pip install tf-keras + fi + # Install tf-keras for tensorflow 2.17.0 + if [ "${{ matrix.tensorflow-version }}" == "2.17.0" ]; then + pip install tf-keras + fi pip install coveralls pip install -r requirements.txt @@ -56,10 +63,8 @@ jobs: print('Python version: ', sys.version); \ import tensorflow as tf; \ import tensorflow_probability as tfp; \ - import tf_keras as tfk; \ print('Tensorflow version =', tf.__version__); \ print('Tensorflow Probability version =', tfp.__version__); \ - print('TF_Keras version =', tfk.__version__); \ " - name: Run unit tests