[doc] Add return value to docstring