From 461f6e84ce94d21ab979447b7382d08d145ded1b Mon Sep 17 00:00:00 2001
From: Da Zheng <zhengda1936@gmail.com>
Date: Sat, 10 Nov 2018 00:30:55 +0000
Subject: [PATCH] fix a bug in dgl_graph_compact.

---
 src/operator/dgl_graph.cc | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/operator/dgl_graph.cc b/src/operator/dgl_graph.cc
index f9bdad871429..c9742bd68775 100644
--- a/src/operator/dgl_graph.cc
+++ b/src/operator/dgl_graph.cc
@@ -481,9 +481,7 @@ NNVM_REGISTER_OP(_contrib_dgl_graph_compact)
 .set_attr<nnvm::FInferType>("FInferType", SubgraphCompactType)
 .set_attr<FComputeEx>("FComputeEx<cpu>", SubgraphCompactComputeExCPU)
 .set_attr<std::string>("key_var_num_args", "num_args")
-.add_argument("graph", "NDArray-or-Symbol[]", "Input graphs.")
-.add_argument("data", "NDArray-or-Symbol[]",
-              "The input arrays that indicate vertex Ids in the graphs.")
+.add_argument("graph_data", "NDArray-or-Symbol[]", "Input graphs and input vertex Ids.")
 .add_arguments(SubgraphCompactParam::__FIELDS__());
 
 }  // namespace op