Bump tensorflow from 2.0.0 to 2.0.1 in /tests (#549)

Bumps [tensorflow](https://github.com/tensorflow/tensorflow) from 2.0.0 to 2.0.1.
- [Release notes](https://github.com/tensorflow/tensorflow/releases)
- [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md)
- [Commits](tensorflow/tensorflow@v2.0.0...v2.0.1)

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

tests/requirements.txt

@@ -2,7 +2,7 @@ torch==1.4.0
 torchvision==0.5.0
 jaxlib==0.1.46
 jax==0.1.66
-tensorflow==2.0.0
+tensorflow==2.0.1
 numba==0.47.0
 matplotlib==3.2.1
 pillow==7.1.2