pin keras~=3.0 in CI

henrysky · Jan 3, 2024 · 5638b14 · 5638b14
1 parent 0baea1e
commit 5638b14
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/ci_tests.yml b/.github/workflows/ci_tests.yml
@@ -27,7 +27,8 @@ jobs:
       - name: Install dependencies
         run: |
           pip install ${{ matrix.BACKEND }}~=${{ matrix.BACKEND_VER }}
-          pip install keras coveralls pydot graphviz pytest pytest-cov
+          pip install keras~=3.0
+          pip install coveralls pydot graphviz pytest pytest-cov
           pip install .
       - name: Cache test data
         uses: actions/cache@v3

diff --git a/astroNN/nn/layers.py b/astroNN/nn/layers.py
@@ -1,6 +1,9 @@
 import math
 import keras
-from keras.layers.input_spec import InputSpec
+try:
+    from keras.layers import InputSpec
+except ImportError:
+    from keras.layers.input_spec import InputSpec
 
 epsilon = keras.backend.epsilon
 initializers = keras.initializers