Browse Source

validate train_id before getting dataset_graph

tags/v1.1.0
luopengting 5 years ago
parent
commit
3eb45eff84
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindinsight/backend/lineagemgr/lineage_api.py

+ 2
- 0
mindinsight/backend/lineagemgr/lineage_api.py View File

@@ -21,6 +21,7 @@ from mindinsight.conf import settings
from mindinsight.datavisual.utils.tools import get_train_id
from mindinsight.datavisual.data_transform.data_manager import DATA_MANAGER
from mindinsight.lineagemgr.cache_item_updater import update_lineage_object
from mindinsight.lineagemgr.common.validator.validate import validate_train_id
from mindinsight.lineagemgr.model import filter_summary_lineage
from mindinsight.utils.exceptions import MindInsightException, ParamValueError

@@ -116,6 +117,7 @@ def get_dataset_graph():
"""

train_id = get_train_id(request)
validate_train_id(train_id)
search_condition = {
'summary_dir': {
'in': [train_id]


Loading…
Cancel
Save