using to_homogeneous instead of the old deprecated name. (#173)

choderalab · Jul 14, 2023 · 7165038 · 7165038
1 parent 5b5fb04
commit 7165038
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espaloma/nn/sequential.py b/espaloma/nn/sequential.py
@@ -135,7 +135,7 @@ def forward(self, g, x=None):
         import dgl
 
         # get homogeneous subgraph
-        g_ = dgl.to_homo(g.edge_type_subgraph(["n1_neighbors_n1"]))
+        g_ = dgl.to_homogeneous(g.edge_type_subgraph(["n1_neighbors_n1"]))
 
         if x is None:
             # get node attributes