don't specify keras

secondmind-labs · Jun 18, 2024 · ae1735b · ae1735b
1 parent c174d6e
commit ae1735b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/.github/workflows/quality-check.yaml b/.github/workflows/quality-check.yaml
@@ -58,7 +58,6 @@ jobs:
               keras: "~=2.15.0"
               tensorflow-probability: "~=0.23.0"
             - tensorflow: "~=2.16.0"
-              keras: "~=2.16.0"
               tensorflow-probability: "~=0.24.0"
           exclude:
             # These older versions of TensorFlow don't work with Python 3.10:

diff --git a/tests/integration/test_compilation.py b/tests/integration/test_compilation.py
@@ -16,7 +16,6 @@
 import numpy as np
 import pytest
 import tensorflow as tf
-from tensorflow import keras
 
 from gpflow.keras import tf_keras
 from gpflow.kernels import RBF
@@ -66,7 +65,7 @@ def build_keras_functional_deep_gp(layer_sizes, num_data):
     gp_layers = build_gp_layers(layer_sizes, num_data)
     likelihood = Gaussian()
 
-    inputs = keras.Input(shape=(layer_sizes[0]))
+    inputs = tf_keras.Input(shape=(layer_sizes[0]))
     x = inputs
     for gp in gp_layers:
         x = gp(x)