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

query getMLModelGroup($urn: String!) {
    mlModelGroup(urn: $urn) {
        ...nonRecursiveMLModelGroupFields
        incoming: relationships(
            input: {
                types: ["DownstreamOf", "Consumes", "Produces", "TrainedBy", "MemberOf"]
                direction: INCOMING
                start: 0
                count: 100
            }
        ) {
            ...fullRelationshipResults
        }
        outgoing: relationships(
            input: {
                types: ["DownstreamOf", "Consumes", "Produces", "TrainedBy", "MemberOf"]
                direction: OUTGOING
                start: 0
                count: 100
            }
        ) {
            ...fullRelationshipResults
        }
    }
}