Add tensorflow_datasets to test dependencies.
Add tensorflow_datasets to test dependencies.

Travis is failing without this dependency:
https://travis-ci.org/google/edward2/builds/616862160

PiperOrigin-RevId: 282424226

google/edward2