espaloma.mm.geometry.geometry_in_graph

espaloma.mm.geometry.geometry_in_graph(g)

Assign values to geometric entities in graphs.

Parameters

g (dgl.DGLHeteroGraph) – Input graph.

Returns

g – Output graph.

Return type

dgl.DGLHeteroGraph

Notes

This function modifies graphs in-place.