提交 31f1d295 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!604 GeneratorDataset column names support string type

Merge pull request !604 from yanghaitao/yht_generatordataset_col_list
......@@ -2569,6 +2569,8 @@ class GeneratorDataset(SourceDataset):
# Random accessible input is also iterable
self.source = (lambda: _iter_fn(source, num_samples))
if column_names is not None and not isinstance(column_names, list):
column_names = [column_names]
self.column_names = column_names
if column_types is not None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册