fix a bug in dgl_graph_compact.

eric-haibin-lin · Nov 10, 2018 · 461f6e8 · 461f6e8
1 parent f966cd9
commit 461f6e8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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