p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!887 fix generatordataset shuffle parameter check 

Merge pull request !887 from panfengfeng/panff/check_generatordataset_shuffle
This commit is contained in:
mindspore-ci-bot 2020-04-30 14:50:27 +08:00 committed by Gitee
parent d436f798dc 5e7cef7e3d
commit 4c32d7e6b8
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/dataset/engine/validators.py
View File

@ -570,6 +570,8 @@ def check_generatordataset(method):
check_param_type(nreq_param_int, param_dict, int)
nreq_param_list = ["column_types"]
check_param_type(nreq_param_list, param_dict, list)
nreq_param_bool = ["shuffle"]
check_param_type(nreq_param_bool, param_dict, bool)
num_shards = param_dict.get("num_shards")
shard_id = param_dict.get("shard_id")