minor change on TF Data2Vec test (#17085)

Commit

4 years ago

minor change on TF Data2Vec test (#17085) Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>