enabling ml tabs (#4471)

datahub-project · Mar 23, 2022 · 4c1d049 · 4c1d049
1 parent 4476356
commit 4c1d049
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/datahub-web-react/src/app/entity/mlFeatureTable/profile/MLFeatureTableProfile.tsx b/datahub-web-react/src/app/entity/mlFeatureTable/profile/MLFeatureTableProfile.tsx
@@ -44,11 +44,19 @@ export const MLFeatureTableProfile = ({ urn }: { urn: string }): JSX.Element =>
                 name: TabType.Features,
                 path: TabType.Features.toLowerCase(),
                 content: <MlFeatureTableFeatures features={features} />,
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
             {
                 name: TabType.Sources,
                 path: TabType.Sources.toLowerCase(),
                 content: <SourcesView features={features} />,
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
             {
                 name: TabType.Ownership,
@@ -59,6 +67,10 @@ export const MLFeatureTableProfile = ({ urn }: { urn: string }): JSX.Element =>
                         lastModifiedAt={(ownership && ownership.lastModified?.time) || 0}
                     />
                 ),
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
         ];
     };

diff --git a/datahub-web-react/src/app/entity/mlModel/profile/MLModelProfile.tsx b/datahub-web-react/src/app/entity/mlModel/profile/MLModelProfile.tsx
@@ -41,16 +41,28 @@ export const MLModelProfile = ({ urn }: { urn: string }): JSX.Element => {
                 name: TabType.Summary,
                 path: TabType.Summary.toLowerCase(),
                 content: <MLModelSummary model={model} />,
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
             {
                 name: TabType.Groups,
                 path: TabType.Groups.toLowerCase(),
                 content: <MLModelGroupsTab model={model} />,
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
             {
                 name: TabType.CustomProperties,
                 path: TabType.CustomProperties.toLowerCase(),
                 content: <Properties properties={model?.properties?.customProperties || EMPTY_ARR} />,
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
             // {
             //     name: TabType.Deployments,
@@ -66,6 +78,10 @@ export const MLModelProfile = ({ urn }: { urn: string }): JSX.Element => {
                         lastModifiedAt={model?.ownership?.lastModified?.time || 0}
                     />
                 ),
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
         ];
     };

diff --git a/datahub-web-react/src/app/entity/mlModelGroup/profile/MLModelGroupProfile.tsx b/datahub-web-react/src/app/entity/mlModelGroup/profile/MLModelGroupProfile.tsx
@@ -38,6 +38,10 @@ export const MLModelGroupProfile = ({ urn }: { urn: string }): JSX.Element => {
                         models={group?.['incoming']?.relationships?.map((relationship) => relationship.entity) || []}
                     />
                 ),
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
             {
                 name: TabType.Ownership,
@@ -48,6 +52,10 @@ export const MLModelGroupProfile = ({ urn }: { urn: string }): JSX.Element => {
                         lastModifiedAt={group?.ownership?.lastModified?.time || 0}
                     />
                 ),
+                display: {
+                    visible: (_, _1) => true,
+                    enabled: (_, _1) => true,
+                },
             },
         ];
     };