From 06bc8c19460142b5ecc325d8e4ffa27bc0bc5810 Mon Sep 17 00:00:00 2001
From: Hyejin Yoon <0327jane@gmail.com>
Date: Thu, 16 Jan 2025 08:27:05 +0900
Subject: [PATCH] add back fragments.graphql

---
 datahub-web-react/src/graphql/fragments.graphql | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/datahub-web-react/src/graphql/fragments.graphql b/datahub-web-react/src/graphql/fragments.graphql
index 56b5dc86254097..344b4db26027a0 100644
--- a/datahub-web-react/src/graphql/fragments.graphql
+++ b/datahub-web-react/src/graphql/fragments.graphql
@@ -897,6 +897,10 @@ fragment nonRecursiveMLModel on MLModel {
             key
             value
         }
+        mlModelLineageInfo {
+            trainingJobs
+            downstreamJobs
+        }
     }
     globalTags {
         ...globalTagsFields
@@ -979,6 +983,10 @@ fragment nonRecursiveMLModelGroupFields on MLModelGroup {
     browsePathV2 {
         ...browsePathV2Fields
     }
+    mlModelLineageInfo {
+        trainingJobs
+        downstreamJobs
+    }
 }
 
 fragment platformFields on DataPlatform {