Fix some doctests (#18359)

Commit

3 years ago

Fix some doctests (#18359) * Fix some doctests Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>