dmlc / dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.
http://dgl.ai
Apache License 2.0
13.17k stars 2.99k forks source link

[dev] add command line interface for common utility functions #7469

Open Rhett-Ying opened 2 weeks ago

Rhett-Ying commented 2 weeks ago

🔨Work Item

IMPORTANT:

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

Depending work items or issues

Rhett-Ying commented 2 weeks ago

@frozenbugs what do you think about this feature request?