diff --git a/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx b/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx index 7f74d52077..dffe59727f 100644 --- a/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx +++ b/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx @@ -10,6 +10,7 @@ import ViewComfyOutlinedIcon from '@mui/icons-material/ViewComfyOutlined'; import TableauWorkbookLogo from '../../../../images/tableau-workbook.svg?react'; import TableauEmbeddedDataSourceLogo from '../../../../images/tableau-embedded-data-source.svg?react'; import TableauPublishedDataSourceLogo from '../../../../images/tableau-published-data-source.svg?react'; +import VertexAIPipelineTaskLogo from '../../../../images/vertexai-pipeline-task.svg?react'; export enum SubType { Database = 'Database', @@ -25,6 +26,7 @@ export enum SubType { LookerExplore = 'Explore', Looker = 'Look', DbtSource = 'Source', + VertexAIPipelineTask = 'Pipeline Task', } export const TYPE_ICON_CLASS_NAME = 'typeIcon'; @@ -62,5 +64,8 @@ export function getSubTypeIcon(subType?: string): JSX.Element | undefined { if (lowerSubType === SubType.TableauWorksheet.toLowerCase()) { return ; } + if (lowerSubType === SubType.VertexAIPipelineTask.toLowerCase()) { + return ; + } return undefined; } diff --git a/datahub-web-react/src/images/vertexai-pipeline-task.svg b/datahub-web-react/src/images/vertexai-pipeline-task.svg new file mode 100644 index 0000000000..660e0db5df --- /dev/null +++ b/datahub-web-react/src/images/vertexai-pipeline-task.svg @@ -0,0 +1,3 @@ + + +