From b191a04a0c4cc3425aa801bc69efe55ce119134c Mon Sep 17 00:00:00 2001 From: liangyongxiong Date: Fri, 29 May 2020 17:06:35 +0800 Subject: [PATCH] add paramater validation for train job caches api --- mindinsight/backend/datavisual/task_manager_api.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindinsight/backend/datavisual/task_manager_api.py b/mindinsight/backend/datavisual/task_manager_api.py index d32d3e8..024b790 100644 --- a/mindinsight/backend/datavisual/task_manager_api.py +++ b/mindinsight/backend/datavisual/task_manager_api.py @@ -87,7 +87,9 @@ def cache_train_jobs(): if data is None: raise ParamMissError('train_ids') - train_ids = data.get('train_ids', []) + train_ids = data.get('train_ids') + if train_ids is None: + raise ParamMissError('train_ids') processor = TrainTaskManager(DATA_MANAGER) cache_result = processor.cache_train_jobs(train_ids) -- GitLab