From cf4506c1c11a182ee2fa4179d2c7ea44c91cae84 Mon Sep 17 00:00:00 2001
From: Bosco Yung <15840328+bhky@users.noreply.github.com>
Date: Fri, 1 Dec 2023 17:28:52 +0900
Subject: [PATCH] Install also tensorflow when testing with JAX, because it's
 required by Keras

---
 .github/workflows/ci.yml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index da68cef..7514b81 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -11,9 +11,9 @@ jobs:
         backend: [ tensorflow, jax ]
         include:
           - backend: tensorflow
-            package: tensorflow
+            packages: tensorflow
           - backend: jax
-            package: jax[cpu]
+            packages: tensorflow jax[cpu]
     name: python-${{ matrix.python }}-${{ matrix.backend }}
     steps:
       - name: Checkout
@@ -26,7 +26,7 @@ jobs:
         run: |
           python3 -m pip install pylint
           python3 -m pip install mypy
-          python3 -m pip install ${{ matrix.package }}
+          python3 -m pip install ${{ matrix.packages }}
       - name: Test
         run: |
           ./tests/install_from_local_and_test.sh ${{ matrix.backend }}
\ No newline at end of file