diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index dbe8e47d03..4f1bb2c2d7 100644 --- a/mindspore/dataset/engine/validators.py +++ b/mindspore/dataset/engine/validators.py @@ -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")