提交 bfecfc9f 编写于 作者: H heleiwang

Fix comment errors

Add parameter verification
上级 3f094393
......@@ -192,7 +192,7 @@ class GraphData:
Args:
node_list (list or numpy.ndarray): The given list of nodes.
feature_types (list or ndarray): The given list of feature types.
feature_types (list or numpy.ndarray): The given list of feature types.
Returns:
numpy.ndarray: array of features.
......@@ -221,7 +221,7 @@ class GraphData:
Args:
edge_list (list or numpy.ndarray): The given list of edges.
feature_types (list or ndarray): The given list of feature types.
feature_types (list or numpy.ndarray): The given list of feature types.
Returns:
numpy.ndarray: array of features.
......@@ -249,7 +249,7 @@ class GraphData:
the feature information of nodes, the number of edges, the type of edges, and the feature information of edges.
Returns:
Dict: Meta information of the graph. The key is node_type, edge_type, node_num, edge_num,
dict: Meta information of the graph. The key is node_type, edge_type, node_num, edge_num,
node_feature_type and edge_feature_type.
"""
return self._graph.graph_info()
......
......@@ -1013,6 +1013,7 @@ def check_gnn_random_walk(method):
type_check(step_home_param, (float,), "step_home_param")
type_check(step_away_param, (float,), "step_away_param")
type_check(default_node, (int,), "default_node")
check_value(default_node, (-1, INT32_MAX), "default_node")
return method(self, *args, **kwargs)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册