!689 Check num_shards and shard_id in TextFileDataset

Merge pull request !689 from yanghaitao/yht_tffileddataset_shard
5 years ago · 86075dc9cb
parent da5b10b699 60f2663326
commit 86075dc9cb
1 changed files with 2 additions and 0 deletions
--- a/mindspore/dataset/engine/validators.py
+++ b/mindspore/dataset/engine/validators.py
@ -906,6 +906,8 @@ def check_textfiledataset(method):

        check_param_type(nreq_param_int, param_dict, int)

+        check_sampler_shuffle_shard_options(param_dict)
+
        return method(*args, **kwargs)

    return new_method