提交 e95220d8 编写于 作者: Y Yanjun Peng

fix generatordataset para check

上级 6e7a38ac
...@@ -665,6 +665,8 @@ def check_generatordataset(method): ...@@ -665,6 +665,8 @@ def check_generatordataset(method):
# check column_names or schema; required argument # check column_names or schema; required argument
column_names = param_dict.get('column_names') column_names = param_dict.get('column_names')
if column_names is not None:
check_columns(column_names, "column_names")
schema = param_dict.get('schema') schema = param_dict.get('schema')
if column_names is None and schema is None: if column_names is None and schema is None:
raise ValueError("Neither columns_names not schema are provided.") raise ValueError("Neither columns_names not schema are provided.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册