feat(client-sagemaker): This release introduces a new optional parame…

…ter: InferenceAmiVersion, in ProductionVariant.
aws · Jun 7, 2024 · 1aa3572 · 1aa3572
1 parent 41f4cc5
commit 1aa3572
Show file tree

Hide file tree

Showing 10 changed files with 165 additions and 106 deletions.
diff --git a/clients/client-sagemaker/src/commands/CreateEndpointConfigCommand.ts b/clients/client-sagemaker/src/commands/CreateEndpointConfigCommand.ts
@@ -96,6 +96,7 @@ export interface CreateEndpointConfigCommandOutput extends CreateEndpointConfigO
  *       RoutingConfig: { // ProductionVariantRoutingConfig
  *         RoutingStrategy: "LEAST_OUTSTANDING_REQUESTS" || "RANDOM", // required
  *       },
+ *       InferenceAmiVersion: "al2-ami-sagemaker-inference-gpu-2",
  *     },
  *   ],
  *   DataCaptureConfig: { // DataCaptureConfig
@@ -208,6 +209,7 @@ export interface CreateEndpointConfigCommandOutput extends CreateEndpointConfigO
  *       RoutingConfig: {
  *         RoutingStrategy: "LEAST_OUTSTANDING_REQUESTS" || "RANDOM", // required
  *       },
+ *       InferenceAmiVersion: "al2-ami-sagemaker-inference-gpu-2",
  *     },
  *   ],
  *   ExecutionRoleArn: "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/CreateSpaceCommand.ts b/clients/client-sagemaker/src/commands/CreateSpaceCommand.ts
@@ -5,8 +5,7 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { CreateSpaceRequest } from "../models/models_1";
-import { CreateSpaceResponse } from "../models/models_2";
+import { CreateSpaceRequest, CreateSpaceResponse } from "../models/models_2";
 import { de_CreateSpaceCommand, se_CreateSpaceCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/DescribeEndpointConfigCommand.ts b/clients/client-sagemaker/src/commands/DescribeEndpointConfigCommand.ts
@@ -73,6 +73,7 @@ export interface DescribeEndpointConfigCommandOutput extends DescribeEndpointCon
  * //       RoutingConfig: { // ProductionVariantRoutingConfig
  * //         RoutingStrategy: "LEAST_OUTSTANDING_REQUESTS" || "RANDOM", // required
  * //       },
+ * //       InferenceAmiVersion: "al2-ami-sagemaker-inference-gpu-2",
  * //     },
  * //   ],
  * //   DataCaptureConfig: { // DataCaptureConfig
@@ -180,6 +181,7 @@ export interface DescribeEndpointConfigCommandOutput extends DescribeEndpointCon
  * //       RoutingConfig: {
  * //         RoutingStrategy: "LEAST_OUTSTANDING_REQUESTS" || "RANDOM", // required
  * //       },
+ * //       InferenceAmiVersion: "al2-ami-sagemaker-inference-gpu-2",
  * //     },
  * //   ],
  * //   ExecutionRoleArn: "STRING_VALUE",

diff --git a/clients/client-sagemaker/src/commands/DescribeProcessingJobCommand.ts b/clients/client-sagemaker/src/commands/DescribeProcessingJobCommand.ts
@@ -5,8 +5,7 @@ import { Command as $Command } from "@smithy/smithy-client";
 import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { commonParams } from "../endpoint/EndpointParameters";
-import { DescribeProcessingJobRequest } from "../models/models_2";
-import { DescribeProcessingJobResponse } from "../models/models_3";
+import { DescribeProcessingJobRequest, DescribeProcessingJobResponse } from "../models/models_3";
 import { de_DescribeProcessingJobCommand, se_DescribeProcessingJobCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/models/models_1.ts b/clients/client-sagemaker/src/models/models_1.ts
@@ -2483,6 +2483,20 @@ export interface ProductionVariantCoreDumpConfig {
   KmsKeyId?: string;
 }
 
+/**
+ * @public
+ * @enum
+ */
+export const ProductionVariantInferenceAmiVersion = {
+  AL2_GPU_2: "al2-ami-sagemaker-inference-gpu-2",
+} as const;
+
+/**
+ * @public
+ */
+export type ProductionVariantInferenceAmiVersion =
+  (typeof ProductionVariantInferenceAmiVersion)[keyof typeof ProductionVariantInferenceAmiVersion];
+
 /**
  * @public
  * @enum
@@ -2706,6 +2720,18 @@ export interface ProductionVariant {
    * @public
    */
   RoutingConfig?: ProductionVariantRoutingConfig;
+
+  /**
+   * <p>Specifies an option from a collection of preconfigured Amazon Machine Image (AMI)
+   *             images. Each image is configured by Amazon Web Services with a set of software and driver
+   *             versions. Amazon Web Services optimizes these configurations for different machine
+   *             learning workloads.</p>
+   *          <p>By selecting an AMI version, you can ensure that your inference environment is
+   *             compatible with specific software requirements, such as CUDA driver versions, Linux
+   *             kernel versions, or Amazon Web Services Neuron driver versions.</p>
+   * @public
+   */
+  InferenceAmiVersion?: ProductionVariantInferenceAmiVersion;
 }
 
 /**
@@ -9859,14 +9885,18 @@ export interface DriftCheckBaselines {
  *             simplified compared to the schema of <code>ModelCard</code>. The
  *             <code>ModelPackageModelCard</code> schema does not include <code>model_package_details</code>,
  *             and <code>model_overview</code> is composed of the <code>model_creator</code> and
- *             <code>model_artifact</code> properties. For more information about
+ *             <code>model_artifact</code> properties. For more information about the model package model
+ *             card schema, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html#model-card-schema">Model
+ *                 package model card schema</a>. For more information about
  *             the model card associated with the model package, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html">View
  *                 the Details of a Model Version</a>.</p>
  * @public
  */
 export interface ModelPackageModelCard {
   /**
-   * <p>The content of the model card.</p>
+   * <p>The content of the model card. The content must follow the schema described
+   *            in <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html#model-card-schema">Model
+   *                Package Model Card Schema</a>.</p>
    * @public
    */
   ModelCardContent?: string;
@@ -10273,7 +10303,9 @@ export interface CreateModelPackageInput {
    *            simplified compared to the schema of <code>ModelCard</code>. The
    *            <code>ModelPackageModelCard</code> schema does not include <code>model_package_details</code>,
    *            and <code>model_overview</code> is composed of the <code>model_creator</code> and
-   *            <code>model_artifact</code> properties. For more information about
+   *            <code>model_artifact</code> properties. For more information about the model package model
+   *            card schema, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html#model-card-schema">Model
+   *                package model card schema</a>. For more information about
    *            the model card associated with the model package, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html">View
    *                the Details of a Model Version</a>.</p>
    * @public
@@ -12300,55 +12332,6 @@ export interface SpaceSharingSettings {
   SharingType: SharingType | undefined;
 }
 
-/**
- * @public
- */
-export interface CreateSpaceRequest {
-  /**
-   * <p>The ID of the associated domain.</p>
-   * @public
-   */
-  DomainId: string | undefined;
-
-  /**
-   * <p>The name of the space.</p>
-   * @public
-   */
-  SpaceName: string | undefined;
-
-  /**
-   * <p>Tags to associated with the space. Each tag consists of a key and an optional value.
-   *       Tag keys must be unique for each resource. Tags are searchable using the
-   *       <code>Search</code> API.</p>
-   * @public
-   */
-  Tags?: Tag[];
-
-  /**
-   * <p>A collection of space settings.</p>
-   * @public
-   */
-  SpaceSettings?: SpaceSettings;
-
-  /**
-   * <p>A collection of ownership settings.</p>
-   * @public
-   */
-  OwnershipSettings?: OwnershipSettings;
-
-  /**
-   * <p>A collection of space sharing settings.</p>
-   * @public
-   */
-  SpaceSharingSettings?: SpaceSharingSettings;
-
-  /**
-   * <p>The name of the space that appears in the SageMaker Studio UI.</p>
-   * @public
-   */
-  SpaceDisplayName?: string;
-}
-
 /**
  * @internal
  */

diff --git a/clients/client-sagemaker/src/models/models_2.ts b/clients/client-sagemaker/src/models/models_2.ts
@@ -146,6 +146,7 @@ import {
   OfflineStoreConfig,
   OnlineStoreConfig,
   OutputConfig,
+  OwnershipSettings,
   ParallelismConfiguration,
   ProcessingInstanceType,
   Processor,
@@ -162,11 +163,62 @@ import {
   ShadowModeConfig,
   SkipModelValidation,
   SourceAlgorithmSpecification,
+  SpaceSettings,
+  SpaceSharingSettings,
   ThroughputMode,
   UserSettings,
   VendorGuidance,
 } from "./models_1";
 
+/**
+ * @public
+ */
+export interface CreateSpaceRequest {
+  /**
+   * <p>The ID of the associated domain.</p>
+   * @public
+   */
+  DomainId: string | undefined;
+
+  /**
+   * <p>The name of the space.</p>
+   * @public
+   */
+  SpaceName: string | undefined;
+
+  /**
+   * <p>Tags to associated with the space. Each tag consists of a key and an optional value.
+   *       Tag keys must be unique for each resource. Tags are searchable using the
+   *       <code>Search</code> API.</p>
+   * @public
+   */
+  Tags?: Tag[];
+
+  /**
+   * <p>A collection of space settings.</p>
+   * @public
+   */
+  SpaceSettings?: SpaceSettings;
+
+  /**
+   * <p>A collection of ownership settings.</p>
+   * @public
+   */
+  OwnershipSettings?: OwnershipSettings;
+
+  /**
+   * <p>A collection of space sharing settings.</p>
+   * @public
+   */
+  SpaceSharingSettings?: SpaceSharingSettings;
+
+  /**
+   * <p>The name of the space that appears in the SageMaker Studio UI.</p>
+   * @public
+   */
+  SpaceDisplayName?: string;
+}
+
 /**
  * @public
  */
@@ -439,17 +491,15 @@ export interface RemoteDebugConfig {
 
 /**
  * <p>Contains information about attribute-based access control (ABAC) for a training job.
- *             The session chaining configuration uses Amazon Security Token Service (STS) for your
- *             training job to request temporary, limited-privilege credentials to tenants. For more
- *             information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-access-training-data.html#model-access-training-data-abac">Attribute-based access control (ABAC) for multi-tenancy
- *                 training</a>.</p>
+ *             The session chaining configuration uses Amazon Security Token Service (STS) for your training job to
+ *             request temporary, limited-privilege credentials to tenants. For more information, see
+ *                 <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-access-training-data.html#model-access-training-data-abac">Attribute-based access control (ABAC) for multi-tenancy training</a>.</p>
  * @public
  */
 export interface SessionChainingConfig {
   /**
-   * <p>Set to <code>True</code> to allow SageMaker to extract session tags from a
-   *             training job creation role and reuse these tags when assuming the training
-   *             job execution role.</p>
+   * <p>Set to <code>True</code> to allow SageMaker to extract session tags from a training job
+   *             creation role and reuse these tags when assuming the training job execution role.</p>
    * @public
    */
   EnableSessionTagChaining?: boolean;
@@ -3820,8 +3870,8 @@ export interface DescribeCompilationJobRequest {
  *          <p>Model artifacts are outputs that result from training a model. They typically consist
  *             of trained parameters, a model definition that describes how to compute inferences, and
  *             other metadata. A SageMaker container stores your trained model artifacts in the
- *                 <code>/opt/ml/model</code> directory. After training has completed, by default, these artifacts
- *             are uploaded to your Amazon S3 bucket as compressed files.</p>
+ *                 <code>/opt/ml/model</code> directory. After training has completed, by default,
+ *             these artifacts are uploaded to your Amazon S3 bucket as compressed files.</p>
  * @public
  */
 export interface ModelArtifacts {
@@ -8926,7 +8976,9 @@ export interface DescribeModelPackageOutput {
    *             simplified compared to the schema of <code>ModelCard</code>. The
    *             <code>ModelPackageModelCard</code> schema does not include <code>model_package_details</code>,
    *             and <code>model_overview</code> is composed of the <code>model_creator</code> and
-   *             <code>model_artifact</code> properties. For more information about
+   *             <code>model_artifact</code> properties. For more information about the model package model
+   *             card schema, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html#model-card-schema">Model
+   *                 package model card schema</a>. For more information about
    *             the model card associated with the model package, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/model-registry-details.html">View
    *                 the Details of a Model Version</a>.</p>
    * @public
@@ -9848,18 +9900,6 @@ export interface DescribePipelineExecutionResponse {
   SelectiveExecutionConfig?: SelectiveExecutionConfig;
 }
 
-/**
- * @public
- */
-export interface DescribeProcessingJobRequest {
-  /**
-   * <p>The name of the processing job. The name must be unique within an Amazon Web Services Region in the
-   *             Amazon Web Services account.</p>
-   * @public
-   */
-  ProcessingJobName: string | undefined;
-}
-
 /**
  * @internal
  */

diff --git a/clients/client-sagemaker/src/models/models_3.ts b/clients/client-sagemaker/src/models/models_3.ts
@@ -147,6 +147,18 @@ import {
   WorkerAccessConfiguration,
 } from "./models_2";
 
+/**
+ * @public
+ */
+export interface DescribeProcessingJobRequest {
+  /**
+   * <p>The name of the processing job. The name must be unique within an Amazon Web Services Region in the
+   *             Amazon Web Services account.</p>
+   * @public
+   */
+  ProcessingJobName: string | undefined;
+}
+
 /**
  * @public
  * @enum
@@ -11178,17 +11190,3 @@ export const NotebookInstanceSortKey = {
  * @public
  */
 export type NotebookInstanceSortKey = (typeof NotebookInstanceSortKey)[keyof typeof NotebookInstanceSortKey];
-
-/**
- * @public
- * @enum
- */
-export const NotebookInstanceSortOrder = {
-  ASCENDING: "Ascending",
-  DESCENDING: "Descending",
-} as const;
-
-/**
- * @public
- */
-export type NotebookInstanceSortOrder = (typeof NotebookInstanceSortOrder)[keyof typeof NotebookInstanceSortOrder];