From bfecfc9f96e57f5d82586224eb43ba8bfd8143bb Mon Sep 17 00:00:00 2001 From: heleiwang Date: Mon, 20 Jul 2020 10:33:55 +0800 Subject: [PATCH] Fix comment errors Add parameter verification --- mindspore/dataset/engine/graphdata.py | 6 +++--- mindspore/dataset/engine/validators.py | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/mindspore/dataset/engine/graphdata.py b/mindspore/dataset/engine/graphdata.py index 81314b4373..3ebca7556e 100644 --- a/mindspore/dataset/engine/graphdata.py +++ b/mindspore/dataset/engine/graphdata.py @@ -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() diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index 99826854ca..ff1ddab9b0 100644 --- a/mindspore/dataset/engine/validators.py +++ b/mindspore/dataset/engine/validators.py @@ -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)