cloudpickle>=1.2.2
dill
mock>=3.0.5
pytest-xdist
virtualenv
dm-tree>=0.1.1
dataclasses==0.7 ; python_version<'3.7'
tf-nightly==2.3.0.dev20200609
optax==0.0.1

cloudpickle>=1.2.2
dill
mock>=3.0.5
pytest-xdist
virtualenv
dm-tree>=0.1.1
dataclasses==0.7 ; python_version<'3.7'
tensorflow==2.3.1
optax==0.0.1

Replace tf-nightly with tensorflow==2.3.1 for testing.
Replace tf-nightly with tensorflow==2.3.1 for testing.

PiperOrigin-RevId: 338646654
Change-Id: I1178943472f197cf940ccbe860bba4139a84641f
