From e95220d89d66fd7ed9d26b9df080a399b9783184 Mon Sep 17 00:00:00 2001 From: Yanjun Peng Date: Wed, 17 Jun 2020 14:52:23 +0800 Subject: [PATCH] fix generatordataset para check --- mindspore/dataset/engine/validators.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/dataset/engine/validators.py b/mindspore/dataset/engine/validators.py index d32a00e175..f924acb57a 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.")