diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index d32a00e1752731825f542ad6d655cd465b700ba4..f924acb57ac660dda604f48d9c3bc440f3186e8b 100644 --- a/mindspore/dataset/engine/validators.py +++ b/mindspore/dataset/engine/validators.py @@ -665,6 +665,8 @@ def check_generatordataset(method): # check column_names or schema; required argument column_names = param_dict.get('column_names') + if column_names is not None: + check_columns(column_names, "column_names") schema = param_dict.get('schema') if column_names is None and schema is None: raise ValueError("Neither columns_names not schema are provided.")