add paramater validation for train job caches api

b191a04a · liangyongxiong · 04bec840 · b191a04a
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

mindinsight/backend/datavisual/task_manager_api.py mindinsight/backend/datavisual/task_manager_api.py +3 -1

未找到文件。
--- 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)