Fix/Update for doctest (#30216)

Commit

1 year ago

Fix/Update for doctest (#30216) fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>