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

query getMLModel($urn: String!) {
    mlModel(urn: $urn) {
        ...nonRecursiveMLModel
        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
        }
    }
}