Bump versions for Tensorflow as needed for Reverb and TFP.
@@ -28,12 +28,11 @@ spec.loader.exec_module(_metadata)
reverb_requirements = [
'dm-reverb',
- 'tensorflow>=2.3.0',
- 'tensorflow<2.4.0',
+ 'tensorflow>=2.4.0',
]
tf_requirements = [
- 'tensorflow>=2.3.0',
+ 'tensorflow>=2.4.0',
'tensorflow_probability',
'dm-sonnet',
'trfl',
