Please add a "Return" section to the docstring that contains a description of the structure of the returned dictionary that lists the keys of the nodes and the expected type and shape of the values.