提交 b33fd4bf 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!211 check if parameter is missing for train job caches api

Merge pull request !211 from liangyongxiong/fix-cache-api
...@@ -25,6 +25,7 @@ from flask import request ...@@ -25,6 +25,7 @@ from flask import request
from flask import jsonify from flask import jsonify
from mindinsight.conf import settings from mindinsight.conf import settings
from mindinsight.utils.exceptions import ParamMissError
from mindinsight.datavisual.common.validation import Validation from mindinsight.datavisual.common.validation import Validation
from mindinsight.datavisual.data_transform.summary_watcher import SummaryWatcher from mindinsight.datavisual.data_transform.summary_watcher import SummaryWatcher
from mindinsight.datavisual.utils.tools import str_to_bool from mindinsight.datavisual.utils.tools import str_to_bool
...@@ -83,6 +84,9 @@ def query_train_jobs(): ...@@ -83,6 +84,9 @@ def query_train_jobs():
def cache_train_jobs(): def cache_train_jobs():
""" Cache train jobs.""" """ Cache train jobs."""
data = request.get_json(silent=True) data = request.get_json(silent=True)
if data is None:
raise ParamMissError('train_ids')
train_ids = data.get('train_ids', []) train_ids = data.get('train_ids', [])
processor = TrainTaskManager(DATA_MANAGER) processor = TrainTaskManager(DATA_MANAGER)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册