Add tf_keras to requirements, for unit testing

This commit is contained in:
Francois Chollet 2024-07-14 21:18:58 -07:00
parent 040bf7755e
commit 1e1007099f

@ -1,6 +1,7 @@
# Tensorflow. # Tensorflow.
tensorflow-cpu~=2.16.2;sys_platform != 'darwin' # Pin to TF 2.16 tensorflow-cpu~=2.16.2;sys_platform != 'darwin' # Pin to TF 2.16
tensorflow~=2.16.2;sys_platform == 'darwin' tensorflow~=2.16.2;sys_platform == 'darwin'
tf_keras
# Torch. # Torch.
# TODO: Pin to < 2.3.0 (GitHub issue #19602) # TODO: Pin to < 2.3.0 (GitHub issue #19602)