From 763aa1067e58ba146b6543bc88de4ccf76b8f912 Mon Sep 17 00:00:00 2001 From: ms_yan <6576637+ms_yan@user.noreply.gitee.com> Date: Wed, 22 Apr 2020 17:08:17 +0800 Subject: [PATCH] Add Error catch for giving too many input parameters --- mindspore/dataset/engine/validators.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index 29bce25bd1..dabeb2d424 100644 --- a/mindspore/dataset/engine/validators.py +++ b/mindspore/dataset/engine/validators.py @@ -233,8 +233,13 @@ def make_param_dict(method, args, kwargs): params = sig.parameters keys = list(params.keys()) param_dict = dict() - for name, value in enumerate(args): - param_dict[keys[name]] = value + try: + for name, value in enumerate(args): + param_dict[keys[name]] = value + except IndexError: + raise TypeError("{0}() expected {1} arguments, but {2} were given".format( + method.__name__, len(keys) - 1, len(args) - 1)) + param_dict.update(zip(params.keys(), args)) param_dict.update(kwargs)