datahub/datahub-web-react/src/graphql/mlModelGroup.graphql

query getMLModelGroup($urn: String!) {
    mlModelGroup(urn: $urn) {
        ...nonRecursiveMLModelGroupFields
        downstreamLineage {
            ...downstreamRelationshipFields
        }
        upstreamLineage {
            ...upstreamRelationshipFields
        }
    }
}