[Transform] [Doc] Rename transform to transforms and update doc (#3765)

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

python/dgl/transform/functional.py → python/dgl/transforms/functional.py

@@ -1042,7 +1042,7 @@ def khop_graph(g, k, copy_ndata=True):
 
     >>> import dgl
     >>> g = dgl.graph(([0, 1], [1, 2]))
-    >>> g_2 = dgl.transform.khop_graph(g, 2)
+    >>> g_2 = dgl.transforms.khop_graph(g, 2)
     >>> print(g_2.edges())
     (tensor([0]), tensor([2]))
 

python/dgl/transform/module.py → python/dgl/transforms/module.py

@@ -671,7 +671,7 @@ class Compose(BaseTransform):
     -------
 
     >>> import dgl
-    >>> from dgl import transform as T
+    >>> from dgl import transforms as T
 
     >>> g = dgl.graph(([0, 0], [1, 1]))
     >>> transform = T.Compose([T.ToSimple(), T.AddReverse()])

tests/compute/test_transform.py

@@ -530,7 +530,7 @@ def get_nodeflow(g, node_ids, num_layers):
 def test_partition_with_halo():
     g = create_large_graph(1000)
     node_part = np.random.choice(4, g.number_of_nodes())
-    subgs, _, _ = dgl.transform.partition_graph_with_halo(g, node_part, 2, reshuffle=True)
+    subgs, _, _ = dgl.transforms.partition_graph_with_halo(g, node_part, 2, reshuffle=True)
     for part_id, subg in subgs.items():
         node_ids = np.nonzero(node_part == part_id)[0]
         lnode_ids = np.nonzero(F.asnumpy(subg.ndata['inner_node']))[0]
@@ -562,7 +562,7 @@ def check_metis_partition_with_constraint(g):
     ntypes = np.zeros((g.number_of_nodes(),), dtype=np.int32)
     ntypes[0:int(g.number_of_nodes()/4)] = 1
     ntypes[int(g.number_of_nodes()*3/4):] = 2
-    subgs = dgl.transform.metis_partition(g, 4, extra_cached_hops=1, balance_ntypes=ntypes)
+    subgs = dgl.transforms.metis_partition(g, 4, extra_cached_hops=1, balance_ntypes=ntypes)
     if subgs is not None:
         for i in subgs:
             subg = subgs[i]
@@ -571,7 +571,7 @@ def check_metis_partition_with_constraint(g):
             print('type0:', np.sum(sub_ntypes == 0))
             print('type1:', np.sum(sub_ntypes == 1))
             print('type2:', np.sum(sub_ntypes == 2))
-    subgs = dgl.transform.metis_partition(g, 4, extra_cached_hops=1,
+    subgs = dgl.transforms.metis_partition(g, 4, extra_cached_hops=1,
                                           balance_ntypes=ntypes, balance_edges=True)
     if subgs is not None:
         for i in subgs:
@@ -583,7 +583,7 @@ def check_metis_partition_with_constraint(g):
             print('type2:', np.sum(sub_ntypes == 2))
 
 def check_metis_partition(g, extra_hops):
-    subgs = dgl.transform.metis_partition(g, 4, extra_cached_hops=extra_hops)
+    subgs = dgl.transforms.metis_partition(g, 4, extra_cached_hops=extra_hops)
     num_inner_nodes = 0
     num_inner_edges = 0
     if subgs is not None:
@@ -600,7 +600,7 @@ def check_metis_partition(g, extra_hops):
         return
 
     # partitions with node reshuffling
-    subgs = dgl.transform.metis_partition(g, 4, extra_cached_hops=extra_hops, reshuffle=True)
+    subgs = dgl.transforms.metis_partition(g, 4, extra_cached_hops=extra_hops, reshuffle=True)
     num_inner_nodes = 0
     num_inner_edges = 0
     edge_cnts = np.zeros((g.number_of_edges(),))