Browse Source

!581 dataset: Add Error catch for giving too many input parameters

Merge pull request !581 from ms_yan/para_num_check
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
faee20726f
1 changed files with 7 additions and 2 deletions
  1. +7
    -2
      mindspore/dataset/engine/validators.py

+ 7
- 2
mindspore/dataset/engine/validators.py View File

@@ -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)



Loading…
Cancel
Save