Upgrade test dependencies (#2544)

Commit

86 days ago

Upgrade test dependencies (#2544) * bump tensorflow * address CI failure * maybe fix deps * bump granite assertion