diff --git a/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx b/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx index c548e079..a1a34c21 100644 --- a/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx +++ b/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx @@ -292,7 +292,7 @@ function ExecuteConfig() { @@ -419,6 +419,7 @@ function ExecuteConfig() { - + diff --git a/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx b/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx index 7cd6af7a..9e641699 100644 --- a/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx +++ b/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx @@ -126,7 +126,7 @@ function AddModelModal({ typeList, tagList, onOk, ...rest }: AddModelModalProps) { - const experimentNode = experimentStatusObjs?.[item.id]; - updateWorkflowNode(item, experimentNode); - }); + if (experimentStatusObjs) { + workflowData.nodes.forEach((item) => { + const experimentNode = experimentStatusObjs?.[item.id]; + updateWorkflowNode(item, experimentNode); + }); + + // 处理workflow状态 + Object.keys(experimentStatusObjs).some((key) => { + if (key.startsWith(NodePrefix)) { + const workflowStatus = experimentStatusObjs[key]; + setWorkflowStatus(workflowStatus); + return true; + } + return false; + }); + } // 绘制图 getGraphData(workflowData, true); @@ -147,6 +161,15 @@ function ExperimentText() { status: phase, })); + const workflowStatus = Object.values(nodes).find((node) => + node.displayName.startsWith(NodePrefix), + ); + + // 设置工作流状态 + if (workflowStatus) { + setWorkflowStatus(workflowStatus); + } + const workflowData = workflowRef.current; workflowData.nodes.forEach((item) => { const experimentNode = Object.values(nodes).find((node) => node.displayName === item.id); @@ -471,13 +494,13 @@ function ExperimentText() {
- 启动时间:{formatDate(experimentIns?.create_time)} + 启动时间:{formatDate(workflowStatus?.startedAt)}
执行时长:
@@ -488,11 +511,11 @@ function ExperimentText() { height: '8px', borderRadius: '50%', marginRight: '6px', - backgroundColor: experimentStatusInfo[experimentIns?.status]?.color, + backgroundColor: experimentStatusInfo[workflowStatus?.phase]?.color, }} >
- - {experimentStatusInfo[experimentIns?.status]?.label} + + {experimentStatusInfo[workflowStatus?.phase]?.label}