From 80ffdf06caa933624ffb11743c1d6e9362c2264b Mon Sep 17 00:00:00 2001 From: liangyongxiong Date: Thu, 28 May 2020 22:15:07 +0800 Subject: [PATCH] check if parameter is missing for train job caches api --- mindinsight/backend/datavisual/task_manager_api.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindinsight/backend/datavisual/task_manager_api.py b/mindinsight/backend/datavisual/task_manager_api.py index f817c48..d32d3e8 100644 --- a/mindinsight/backend/datavisual/task_manager_api.py +++ b/mindinsight/backend/datavisual/task_manager_api.py @@ -25,6 +25,7 @@ from flask import request from flask import jsonify from mindinsight.conf import settings +from mindinsight.utils.exceptions import ParamMissError from mindinsight.datavisual.common.validation import Validation from mindinsight.datavisual.data_transform.summary_watcher import SummaryWatcher from mindinsight.datavisual.utils.tools import str_to_bool @@ -83,6 +84,9 @@ def query_train_jobs(): def cache_train_jobs(): """ Cache train jobs.""" data = request.get_json(silent=True) + if data is None: + raise ParamMissError('train_ids') + train_ids = data.get('train_ids', []) processor = TrainTaskManager(DATA_MANAGER) -- GitLab