Fixing job node color for running and other states.

MarquezProject · Sep 12, 2024 · be64c50 · be64c50
1 parent d407e66
commit be64c50
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/web/src/routes/table-level/TableLineageJobNode.tsx b/web/src/routes/table-level/TableLineageJobNode.tsx
@@ -16,6 +16,7 @@ import MQTooltip from '../../components/core/tooltip/MQTooltip'
 import MqStatus from '../../components/core/status/MqStatus'
 import MqText from '../../components/core/text/MqText'
 import React from 'react'
+import {runStateColor} from "../../helpers/nodes";
 
 interface StateProps {
   lineage: LineageGraph
@@ -84,11 +85,7 @@ const TableLineageJobNode = ({ node }: TableLineageJobNodeProps & StateProps) =>
             </MqText>
             <MqStatus
               label={job.latestRun?.state || 'N/A'}
-              color={
-                job.latestRun?.state === 'COMPLETED'
-                  ? theme.palette.primary.main
-                  : theme.palette.error.main
-              }
+              color={job.latestRun?.state ? runStateColor(job.latestRun?.state) : theme.palette.primary.main}
             />
           </Box>
         </Box>