modify importing module

ZhenyuLU-Heliodore · ZhenyuLU-Heliodore · commit d344e025c876 · 2023-01-07T16:34:45.000+08:00
diff --git a/python/dgl/transforms/functional.py b/python/dgl/transforms/functional.py
@@ -81,6 +81,7 @@
     'to_double',
     'double_radius_node_labeling',
     'shortest_dist',
+    'svd_pe'
     ]
 
 
diff --git a/python/dgl/transforms/module.py b/python/dgl/transforms/module.py
@@ -54,7 +54,8 @@
     'DropEdge',
     'AddEdge',
     'SIGNDiffusion',
-    'ToLevi'
+    'ToLevi',
+    'SvdPE'
 ]
 
 def update_graph_structure(g, data_dict, copy_edata=True):

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,8 @@`
`54`	`54`	`'DropEdge',`
`55`	`55`	`'AddEdge',`
`56`	`56`	`'SIGNDiffusion',`
`57`		`- 'ToLevi'`
	`57`	`+ 'ToLevi',`
	`58`	`+ 'SvdPE'`
`58`	`59`	`]`
`59`	`60`
`60`	`61`	`def update_graph_structure(g, data_dict, copy_edata=True):`