small lineage ui nits

datahub-project · Mar 10, 2022 · 30a5f70 · 30a5f70
1 parent 6e2e76b
commit 30a5f70
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/datahub-web-react/src/app/entity/dataset/DatasetEntity.tsx b/datahub-web-react/src/app/entity/dataset/DatasetEntity.tsx
@@ -111,9 +111,13 @@ export class DatasetEntity implements Entity<Dataset> {
                     component: LineageTab,
                     display: {
                         visible: (_, _1) => true,
-                        enabled: (_, dataset: GetDatasetQuery) =>
-                            (dataset?.dataset?.upstream?.count || 0) > 0 ||
-                            (dataset?.dataset?.downstream?.count || 0) > 0,
+                        enabled: (_, dataset: GetDatasetQuery) => {
+                            console.log(dataset?.dataset?.upstream, dataset?.dataset?.downstream);
+                            return (
+                                (dataset?.dataset?.upstream?.total || 0) > 0 ||
+                                (dataset?.dataset?.downstream?.total || 0) > 0
+                            );
+                        },
                     },
                 },
                 {

diff --git a/datahub-web-react/src/app/lineage/LineageEntityNode.tsx b/datahub-web-react/src/app/lineage/LineageEntityNode.tsx
@@ -261,7 +261,9 @@ export default function LineageEntityNode({
                             |{' '}
                         </tspan>
                         <tspan dx=".25em" dy="-2px">
-                            {capitalizeFirstLetter(node.data.subtype || node.data.type)}
+                            {capitalizeFirstLetter(
+                                node.data.subtype || (node.data.type && entityRegistry.getEntityName(node.data.type)),
+                            )}
                         </tspan>
                     </UnselectableText>
                     {expandTitles ? (