fix more doctests (#22292)

Commit

2 years ago

fix more doctests (#22292) * fix more doctests * fix style --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>