From 5e7cef7e3d4bc732ac4533ccb1c5a9260d99de0a Mon Sep 17 00:00:00 2001 From: panfengfeng Date: Thu, 30 Apr 2020 13:12:05 +0800 Subject: [PATCH] fix generatordataset check shuffle parameter --- mindspore/dataset/engine/validators.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index dbe8e47d031..4f1bb2c2d7b 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")