pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!18249 Fix gnn API log error 

Merge pull request !18249 from lizhenglong1992/gnn_bug_fix
This commit is contained in:
i-robot 2021-06-15 21:05:36 +08:00 committed by Gitee
parent f831272c81 f183dff2b9
commit 07a3ebeef6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/dataset/core/validator_helpers.py
View File

@ -587,13 +587,13 @@ def check_gnn_list_of_pair_or_ndarray(param, param_name):
"""
type_check(param, (list, np.ndarray), param_name)
if isinstance(param, list):
param_names = ["pair_{0}".format(i) for i in range(len(param))]
param_names = ["node_list[{0}]".format(i) for i in range(len(param))]
type_check_list(param, (tuple,), param_names)
for idx, pair in enumerate(param):
if not len(pair) == 2:
raise ValueError("Each member in {0} must be a pair which means length == 2. Got length {1}".format(
param_names[idx], len(pair)))
column_names = ["element_{0}".format(i) for i in range(len(pair))]
column_names = ["node_list[{0}], number #{1} element".format(idx, i+1) for i in range(len(pair))]
type_check_list(pair, (int,), column_names)
elif isinstance(param, np.ndarray):
if param.ndim != 2: