chore(sagemaker): reference interfaces

packages/@aws-cdk-testing/framework-integ/test/aws-scheduler-targets/test/integ.sage-maker-start-pipeline-execution.ts

@@ -94,6 +94,10 @@ class FakePipeline extends cdk.Resource implements sagemaker.IPipeline {
     });
   }
 
+  public get pipelineRef(): cdk.aws_sagemaker.PipelineReference {
+    return { pipelineName: this.pipelineName };
+  }
+
   public grantStartPipelineExecution(grantee: IGrantable): Grant {
     return Grant.addToPrincipal({
       grantee,

packages/@aws-cdk/aws-pipes-targets-alpha/test/integ.sagemaker.ts

@@ -42,6 +42,7 @@ interface FakePipelineProps {
 }
 
 class FakePipeline extends cdk.Resource implements sagemaker.IPipeline {
+  public readonly pipelineRef: sagemaker.PipelineReference;
   public readonly pipelineArn;
   public readonly pipelineName;
 
@@ -125,6 +126,8 @@ class FakePipeline extends cdk.Resource implements sagemaker.IPipeline {
       resourceName: pipeline.pipelineName,
       arnFormat: cdk.ArnFormat.SLASH_RESOURCE_NAME,
     });
+
+    this.pipelineRef = { pipelineName: this.pipelineName };
   }
 
   public grantStartPipelineExecution(grantee: iam.IGrantable): iam.Grant {

packages/@aws-cdk/aws-pipes-targets-alpha/test/sagemaker.test.ts

@@ -2,7 +2,7 @@ import { InputTransformation, Pipe } from '@aws-cdk/aws-pipes-alpha';
 import { App, Stack, Resource } from 'aws-cdk-lib';
 import { Template } from 'aws-cdk-lib/assertions';
 import * as iam from 'aws-cdk-lib/aws-iam';
-import { IPipeline } from 'aws-cdk-lib/aws-sagemaker';
+import { IPipeline, PipelineReference } from 'aws-cdk-lib/aws-sagemaker';
 import { TestSource } from './test-classes';
 import { SageMakerTarget } from '../lib';
 
@@ -131,4 +131,8 @@ class FakePipeline extends Resource implements IPipeline {
       resourceArns: [this.pipelineArn],
     });
   }
+
+  public get pipelineRef(): PipelineReference {
+    return { pipelineName: this.pipelineName };
+  }
 }

packages/@aws-cdk/aws-sagemaker-alpha/lib/endpoint.ts

@@ -304,6 +304,12 @@ abstract class EndpointBase extends cdk.Resource implements IEndpoint {
       resourceArns: [this.endpointArn],
     });
   }
+
+  public get endpointRef(): sagemaker.EndpointReference {
+    return {
+      endpointArn: this.endpointArn,
+    };
+  }
 }
 
 /**

packages/aws-cdk-lib/aws-apigateway/test/integrations/sagemaker.test.ts

@@ -53,4 +53,8 @@ class FakeEndpoint extends cdk.Resource implements sagemaker.IEndpoint {
       resourceArns: [this.endpointArn],
     });
   }
+
+  public get endpointRef(): sagemaker.EndpointReference {
+    return { endpointArn: this.endpointArn };
+  }
 }

packages/aws-cdk-lib/aws-sagemaker/lib/endpoint.ts

@@ -1,10 +1,11 @@
 import * as iam from '../../aws-iam';
 import { IResource } from '../../core';
+import { IEndpointRef } from '../../interfaces/generated/aws-sagemaker-interfaces.generated';
 
 /**
  * The interface for a SageMaker Endpoint resource.
  */
-export interface IEndpoint extends IResource {
+export interface IEndpoint extends IResource, IEndpointRef {
   /**
    * The ARN of the endpoint.
    *

packages/aws-cdk-lib/aws-sagemaker/lib/pipeline.ts

@@ -1,10 +1,11 @@
 import { Grant, IGrantable } from '../../aws-iam';
 import { IResource } from '../../core';
+import { IPipelineRef } from '../../interfaces/generated/aws-sagemaker-interfaces.generated';
 
 /**
  * The interface for a SageMaker Pipeline resource.
  */
-export interface IPipeline extends IResource {
+export interface IPipeline extends IResource, IPipelineRef {
   /**
    * The ARN of the pipeline.
    *

packages/aws-cdk-lib/aws-scheduler-targets/test/sage-maker-start-pipeline-execution.test.ts

@@ -1,6 +1,6 @@
 import { Template } from '../../assertions';
 import { AccountRootPrincipal, Grant, IGrantable, Role } from '../../aws-iam';
-import { IPipeline } from '../../aws-sagemaker';
+import { IPipeline, PipelineReference } from '../../aws-sagemaker';
 import { ScheduleExpression, Schedule, ScheduleGroup } from '../../aws-scheduler';
 import * as sqs from '../../aws-sqs';
 import { App, Duration, Resource, Stack } from '../../core';
@@ -592,6 +592,10 @@ class FakePipeline extends Resource implements IPipeline {
     this.pipelineName = props.pipelineName;
   }
 
+  public get pipelineRef(): PipelineReference {
+    return { pipelineName: this.pipelineName };
+  }
+
   public grantStartPipelineExecution(grantee: IGrantable): Grant {
     return Grant.addToPrincipal({
       grantee,