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)