Upgrade test dependencies (#2544)

* bump tensorflow

* address CI failure

* maybe fix deps

* bump granite assertion

Author: michaeldwan