dgl
dgl copied to clipboard
[dev] add command line interface for common utility functions
🔨Work Item
IMPORTANT:
- This template is only for dev team to track project progress. For feature request or bug report, please use the corresponding issue templates.
- DO NOT create a new work item if the purpose is to fix an existing issue or feature request. We will directly use the issue in the project tracker.
Project tracker: https://github.com/orgs/dmlc/projects/2
Description
For example, when we want to convert DGL partitions into GraphBolt formats, we are required to call like:
python -c "import dgl;dgl.distributed.dgl_partition_to_graphbolt(${DGL_INPUT}, store_eids=True, graph_formats=['coo', 'csc'])"
But when we run such command on AWS batch jobs, it's not easy to escape string properly. A desired way is like:
python -m dgl.distributed.dgl_partition_to_graphbolt --input-path /mount/data/metadata.json --graph-formats coo,csc