diff --git a/mindinsight/backend/datavisual/task_manager_api.py b/mindinsight/backend/datavisual/task_manager_api.py
index d32d3e8589c6ef866b597071bba82647ecb9dc96..024b7907edf162fcc53b42619955abd89c4b75ad 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)