Minor changes for TF2.

3 jobs for dev_tf2 in 34 seconds (queued for 2 seconds)
Status Job ID Name Coverage
  Lint
passed #302948
docker
lint

00:00:16

 
  Test
passed #302949
docker
unittests_py2

00:00:16

passed #302950
docker
unittests_py3

00:00:17