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

query getMLPrimaryKey($urn: String!) {
    mlPrimaryKey(urn: $urn) {
        ...nonRecursiveMLPrimaryKey
        featureTables: relationships(input: { types: ["KeyedBy"], direction: INCOMING, start: 0, count: 100 }) {
            ...fullRelationshipResults
        }
        autoRenderAspects: aspects(input: { autoRenderOnly: true }) {
            ...autoRenderAspectFields
        }
    }
}