[Distributed] Fix the path for distributed partitioning. (#2700)

Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>

examples/pytorch/rgcn/experimental/get_mag_data.py

@@ -48,7 +48,7 @@ for part_id in range(metadata['num_parts']):
         for name in hg.nodes[ntype].data:
             node_data[ntype + '/' + name] = hg.nodes[ntype].data[name][local_nodes]
     print('node features:', node_data.keys())
-    dgl.data.utils.save_tensors(metadata['part-{}'.format(part_id)]['node_feats'], node_data)
+    dgl.data.utils.save_tensors('outputs/' + metadata['part-{}'.format(part_id)]['node_feats'], node_data)
 
     edge_data = {}
     for etype in hg.etypes:
@@ -56,4 +56,4 @@ for part_id in range(metadata['num_parts']):
         for name in hg.edges[etype].data:
             edge_data[etype + '/' + name] = hg.edges[etype].data[name][local_edges]
     print('edge features:', edge_data.keys())
-    dgl.data.utils.save_tensors(metadata['part-{}'.format(part_id)]['edge_feats'], edge_data)
+    dgl.data.utils.save_tensors('outputs/' + metadata['part-{}'.format(part_id)]['edge_feats'], edge_data)

tools/convert_partition.py

@@ -218,7 +218,7 @@ part_metadata = {'graph_name': graph_name,
                  'etypes': etypes_map}
 
 for part_id in range(num_parts):
-    part_dir = output_dir + '/part' + str(part_id)
+    part_dir = 'part' + str(part_id)
     node_feat_file = os.path.join(part_dir, "node_feat.dgl")
     edge_feat_file = os.path.join(part_dir, "edge_feat.dgl")
     part_graph_file = os.path.join(part_dir, "graph.dgl")