From 0b87620b73fa428593b7dc2454cd4518b544e41b Mon Sep 17 00:00:00 2001 From: heleiwang Date: Mon, 25 May 2020 20:07:39 +0800 Subject: [PATCH] fix input check --- mindspore/ccsrc/dataset/engine/gnn/graph.cc | 1 + mindspore/dataset/engine/validators.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/mindspore/ccsrc/dataset/engine/gnn/graph.cc b/mindspore/ccsrc/dataset/engine/gnn/graph.cc index a0278761229..74e7b851537 100644 --- a/mindspore/ccsrc/dataset/engine/gnn/graph.cc +++ b/mindspore/ccsrc/dataset/engine/gnn/graph.cc @@ -152,6 +152,7 @@ Status Graph::GetNodeFeature(const std::shared_ptr &nodes, const std::ve if (!nodes || nodes->Size() == 0) { RETURN_STATUS_UNEXPECTED("Input nodes is empty"); } + CHECK_FAIL_RETURN_UNEXPECTED(!feature_types.empty(), "Inpude feature_types is empty"); TensorRow tensors; for (auto f_type : feature_types) { std::shared_ptr default_feature; diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index abbc15f0fec..049931c80e6 100644 --- a/mindspore/dataset/engine/validators.py +++ b/mindspore/dataset/engine/validators.py @@ -1153,6 +1153,9 @@ def check_aligned_list(param, param_name, membor_type): if not isinstance(param, list): raise TypeError("Parameter {0} is not a list".format(param_name)) + if not param: + raise TypeError( + "Parameter {0} or its members are empty".format(param_name)) membor_have_list = None list_len = None for membor in param: