提交 203b05d4 编写于 作者: Y yanghaitao

num_shards and sampler is not supported if source does not have __getitem__

上级 64b3b566
......@@ -587,6 +587,11 @@ def check_generatordataset(method):
except TypeError:
raise TypeError("sampler should be either iterable or from mindspore.dataset.samplers")
if sampler is not None and not hasattr(source, "__getitem__"):
raise ValueError("sampler is not supported if source does not have attribute '__getitem__'")
if num_shards is not None and not hasattr(source, "__getitem__"):
raise ValueError("num_shards is not supported if source does not have attribute '__getitem__'")
return method(*args, **kwargs)
return new_method
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册