diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql b/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql index b07283885bacc..c3b0243f0695d 100644 --- a/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql +++ b/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql @@ -713,6 +713,12 @@ enum BackfillPolicyType { type AssetDependency { asset: AssetNode! inputName: String! + partitionMapping: PartitionMapping +} + +type PartitionMapping { + name: String! + description: String! } type AssetFreshnessInfo { diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts b/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts index 6d36a6d91fa07..0298376748584 100644 --- a/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts +++ b/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts @@ -253,6 +253,7 @@ export type AssetDependency = { __typename: 'AssetDependency'; asset: AssetNode; inputName: Scalars['String']; + partitionMapping: Maybe; }; export enum AssetEventType { @@ -2453,6 +2454,12 @@ export type PartitionKeys = { export type PartitionKeysOrError = PartitionKeys | PartitionSubsetDeserializationError; +export type PartitionMapping = { + __typename: 'PartitionMapping'; + description: Scalars['String']; + name: Scalars['String']; +}; + export type PartitionRangeSelector = { end: Scalars['String']; start: Scalars['String']; @@ -4894,6 +4901,12 @@ export const buildAssetDependency = ( : buildAssetNode({}, relationshipsToOmit), inputName: overrides && overrides.hasOwnProperty('inputName') ? overrides.inputName! : 'aspernatur', + partitionMapping: + overrides && overrides.hasOwnProperty('partitionMapping') + ? overrides.partitionMapping! + : relationshipsToOmit.has('PartitionMapping') + ? ({} as PartitionMapping) + : buildPartitionMapping({}, relationshipsToOmit), }; }; @@ -9116,6 +9129,22 @@ export const buildPartitionKeys = ( }; }; +export const buildPartitionMapping = ( + overrides?: Partial, + _relationshipsToOmit: Set = new Set(), +): {__typename: 'PartitionMapping'} & PartitionMapping => { + const relationshipsToOmit: Set = new Set(_relationshipsToOmit); + relationshipsToOmit.add('PartitionMapping'); + return { + __typename: 'PartitionMapping', + description: + overrides && overrides.hasOwnProperty('description') + ? overrides.description! + : 'voluptatibus', + name: overrides && overrides.hasOwnProperty('name') ? overrides.name! : 'non', + }; +}; + export const buildPartitionRangeSelector = ( overrides?: Partial, _relationshipsToOmit: Set = new Set(),