recover docstring

Signed-off-by: Will Guo <willg@nvidia.com>

Author: willg-nv