From 87f4023d6df82b1e0e619beda7e2be519f60326c Mon Sep 17 00:00:00 2001 From: shenwei41 Date: Mon, 10 May 2021 09:43:50 +0800 Subject: [PATCH] Modify error info --- mindspore/dataset/core/validator_helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/dataset/core/validator_helpers.py b/mindspore/dataset/core/validator_helpers.py index 4ab8b6c073..dc3f298c63 100644 --- a/mindspore/dataset/core/validator_helpers.py +++ b/mindspore/dataset/core/validator_helpers.py @@ -334,7 +334,7 @@ def check_sampler_shuffle_shard_options(param_dict): raise RuntimeError("sampler and num_samples cannot be specified at the same time.") if num_shards is not None: - check_pos_int32(num_shards) + check_pos_int32(num_shards, "num_shards") if shard_id is None: raise RuntimeError("num_shards is specified and currently requires shard_id as well.") check_value(shard_id, [0, num_shards - 1], "shard_id")