Browse Source

!887 fix generatordataset shuffle parameter check

Merge pull request !887 from panfengfeng/panff/check_generatordataset_shuffle
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
4c32d7e6b8
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/dataset/engine/validators.py

+ 2
- 0
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")


Loading…
Cancel
Save