Add tf_keras to requirements, for unit testing
This commit is contained in:
parent
040bf7755e
commit
1e1007099f
@ -1,6 +1,7 @@
|
||||
# Tensorflow.
|
||||
tensorflow-cpu~=2.16.2;sys_platform != 'darwin' # Pin to TF 2.16
|
||||
tensorflow~=2.16.2;sys_platform == 'darwin'
|
||||
tf_keras
|
||||
|
||||
# Torch.
|
||||
# TODO: Pin to < 2.3.0 (GitHub issue #19602)
|
||||
|
Loading…
Reference in New Issue
Block a user