From d7c4e6f13662568f325012986ffd7a3b7768bbbd Mon Sep 17 00:00:00 2001 From: awstools Date: Thu, 21 Dec 2023 19:17:56 +0000 Subject: [PATCH] feat(client-sagemaker): Amazon SageMaker Training now provides model training container access for debugging purposes. Amazon SageMaker Search now provides the ability to use visibility conditions to limit resource access to a single domain or multiple domains. --- .../src/commands/CreateTrainingJobCommand.ts | 3 + .../commands/DescribeTrainingJobCommand.ts | 3 + .../src/commands/SearchCommand.ts | 6 + .../src/commands/UpdateTrainingJobCommand.ts | 3 + .../client-sagemaker/src/models/models_2.ts | 46 ++-- .../client-sagemaker/src/models/models_3.ts | 188 +++------------ .../client-sagemaker/src/models/models_4.ts | 214 +++++++++++++++++- .../src/protocols/Aws_json1_1.ts | 26 ++- codegen/sdk-codegen/aws-models/sagemaker.json | 109 ++++++++- 9 files changed, 414 insertions(+), 184 deletions(-) diff --git a/clients/client-sagemaker/src/commands/CreateTrainingJobCommand.ts b/clients/client-sagemaker/src/commands/CreateTrainingJobCommand.ts index 10a2160863e62..8e3514ce49ab8 100644 --- a/clients/client-sagemaker/src/commands/CreateTrainingJobCommand.ts +++ b/clients/client-sagemaker/src/commands/CreateTrainingJobCommand.ts @@ -292,6 +292,9 @@ export interface CreateTrainingJobCommandOutput extends CreateTrainingJobRespons * RetryStrategy: { // RetryStrategy * MaximumRetryAttempts: Number("int"), // required * }, + * RemoteDebugConfig: { // RemoteDebugConfig + * EnableRemoteDebug: true || false, + * }, * InfraCheckConfig: { // InfraCheckConfig * EnableInfraCheck: true || false, * }, diff --git a/clients/client-sagemaker/src/commands/DescribeTrainingJobCommand.ts b/clients/client-sagemaker/src/commands/DescribeTrainingJobCommand.ts index ae10441e6b8da..2ccc86cde5ffa 100644 --- a/clients/client-sagemaker/src/commands/DescribeTrainingJobCommand.ts +++ b/clients/client-sagemaker/src/commands/DescribeTrainingJobCommand.ts @@ -276,6 +276,9 @@ export interface DescribeTrainingJobCommandOutput extends DescribeTrainingJobRes * // RetryStrategy: { // RetryStrategy * // MaximumRetryAttempts: Number("int"), // required * // }, + * // RemoteDebugConfig: { // RemoteDebugConfig + * // EnableRemoteDebug: true || false, + * // }, * // InfraCheckConfig: { // InfraCheckConfig * // EnableInfraCheck: true || false, * // }, diff --git a/clients/client-sagemaker/src/commands/SearchCommand.ts b/clients/client-sagemaker/src/commands/SearchCommand.ts index bac89d7f471f8..f98b5f13d4d6e 100644 --- a/clients/client-sagemaker/src/commands/SearchCommand.ts +++ b/clients/client-sagemaker/src/commands/SearchCommand.ts @@ -103,6 +103,12 @@ export interface SearchCommandOutput extends SearchResponse, __MetadataBearer {} * NextToken: "STRING_VALUE", * MaxResults: Number("int"), * CrossAccountFilterOption: "SameAccount" || "CrossAccount", + * VisibilityConditions: [ // VisibilityConditionsList + * { // VisibilityConditions + * Key: "STRING_VALUE", + * Value: "STRING_VALUE", + * }, + * ], * }; * const command = new SearchCommand(input); * const response = await client.send(command); diff --git a/clients/client-sagemaker/src/commands/UpdateTrainingJobCommand.ts b/clients/client-sagemaker/src/commands/UpdateTrainingJobCommand.ts index 8f8f623c1cf8c..f1d126108de24 100644 --- a/clients/client-sagemaker/src/commands/UpdateTrainingJobCommand.ts +++ b/clients/client-sagemaker/src/commands/UpdateTrainingJobCommand.ts @@ -71,6 +71,9 @@ export interface UpdateTrainingJobCommandOutput extends UpdateTrainingJobRespons * ResourceConfig: { // ResourceConfigForUpdate * KeepAlivePeriodInSeconds: Number("int"), // required * }, + * RemoteDebugConfig: { // RemoteDebugConfigForUpdate + * EnableRemoteDebug: true || false, + * }, * }; * const command = new UpdateTrainingJobCommand(input); * const response = await client.send(command); diff --git a/clients/client-sagemaker/src/models/models_2.ts b/clients/client-sagemaker/src/models/models_2.ts index 9e25b371d1ffa..29a4e44cf9c19 100644 --- a/clients/client-sagemaker/src/models/models_2.ts +++ b/clients/client-sagemaker/src/models/models_2.ts @@ -224,6 +224,20 @@ export interface ProfilerRuleConfiguration { RuleParameters?: Record; } +/** + * @public + *

Configuration for remote debugging for the CreateTrainingJob API. To learn more about the remote + * debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) + * for remote debugging.

+ */ +export interface RemoteDebugConfig { + /** + * @public + *

If set to True, enables remote debugging.

+ */ + EnableRemoteDebug?: boolean; +} + /** * @public *

Configuration of storage locations for the Amazon SageMaker Debugger TensorBoard output data.

@@ -476,6 +490,14 @@ export interface CreateTrainingJobRequest { */ RetryStrategy?: RetryStrategy; + /** + * @public + *

Configuration for remote debugging. To learn more about the remote + * debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) + * for remote debugging.

+ */ + RemoteDebugConfig?: RemoteDebugConfig; + /** * @public *

Contains information about the infrastructure health check configuration for the training job.

@@ -10002,30 +10024,6 @@ export interface DescribeTrainingJobRequest { TrainingJobName: string | undefined; } -/** - * @public - *

The name, value, and date and time of a metric that was emitted to Amazon CloudWatch.

- */ -export interface MetricData { - /** - * @public - *

The name of the metric.

- */ - MetricName?: string; - - /** - * @public - *

The value of the metric.

- */ - Value?: number; - - /** - * @public - *

The date and time that the algorithm emitted the metric.

- */ - Timestamp?: Date; -} - /** * @internal */ diff --git a/clients/client-sagemaker/src/models/models_3.ts b/clients/client-sagemaker/src/models/models_3.ts index a1771bd19786d..c691ea1e5295c 100644 --- a/clients/client-sagemaker/src/models/models_3.ts +++ b/clients/client-sagemaker/src/models/models_3.ts @@ -14,18 +14,15 @@ import { AssociationSummary, AutoMLCandidate, AutoMLJobStatus, - AutoMLJobStepMetadata, AutoMLJobSummary, AutoMLSortBy, AutoMLSortOrder, BatchDataCaptureConfig, BatchStrategy, - CallbackStepMetadata, CandidateSortBy, CandidateStatus, Channel, CheckpointConfig, - ClarifyCheckStepMetadata, ClusterNodeSummary, ClusterSortBy, ClusterSummary, @@ -35,7 +32,6 @@ import { CognitoConfig, CompilationJobStatus, CompilationJobSummary, - ConditionStepMetadata, ContextSummary, MetadataProperties, ModelApprovalStatus, @@ -113,7 +109,6 @@ import { LabelingJobStatus, LastUpdateStatus, MemberDefinition, - MetricData, ModelArtifacts, ModelCardExportJobStatus, ModelClientConfig, @@ -130,6 +125,7 @@ import { ProfilerRuleConfiguration, RecommendationJobStatus, RecommendationMetrics, + RemoteDebugConfig, RuleEvaluationStatus, ScheduleStatus, SourceIpConfig, @@ -141,6 +137,30 @@ import { TrialComponentStatus, } from "./models_2"; +/** + * @public + *

The name, value, and date and time of a metric that was emitted to Amazon CloudWatch.

+ */ +export interface MetricData { + /** + * @public + *

The name of the metric.

+ */ + MetricName?: string; + + /** + * @public + *

The value of the metric.

+ */ + Value?: number; + + /** + * @public + *

The date and time that the algorithm emitted the metric.

+ */ + Timestamp?: Date; +} + /** * @public *

Information about the status of the rule evaluation.

@@ -378,9 +398,6 @@ export interface SecondaryStatusTransition { *
*
    *
  • - *

    Downloading the training image.

    - *
    • - *
  • *

    Training * image download completed. Training in * progress.

    @@ -931,6 +948,14 @@ export interface DescribeTrainingJobResponse { */ RetryStrategy?: RetryStrategy; + /** + * @public + *

    Configuration for remote debugging. To learn more about the remote + * debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) + * for remote debugging.

    + */ + RemoteDebugConfig?: RemoteDebugConfig; + /** * @public *

    Contains information about the infrastructure health check configuration for the training job.

    @@ -11091,150 +11116,3 @@ export interface TuningJobStepMetaData { */ Arn?: string; } - -/** - * @public - *

    Metadata for a step execution.

    - */ -export interface PipelineExecutionStepMetadata { - /** - * @public - *

    The Amazon Resource Name (ARN) of the training job that was run by this step execution.

    - */ - TrainingJob?: TrainingJobStepMetadata; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the processing job that was run by this step execution.

    - */ - ProcessingJob?: ProcessingJobStepMetadata; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the transform job that was run by this step execution.

    - */ - TransformJob?: TransformJobStepMetadata; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the tuning job that was run by this step execution.

    - */ - TuningJob?: TuningJobStepMetaData; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the model that was created by this step execution.

    - */ - Model?: ModelStepMetadata; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the model package that the model was registered to by this step execution.

    - */ - RegisterModel?: RegisterModelStepMetadata; - - /** - * @public - *

    The outcome of the condition evaluation that was run by this step execution.

    - */ - Condition?: ConditionStepMetadata; - - /** - * @public - *

    The URL of the Amazon SQS queue used by this step execution, the pipeline generated token, - * and a list of output parameters.

    - */ - Callback?: CallbackStepMetadata; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the Lambda function that was run by this step execution and a list of - * output parameters.

    - */ - Lambda?: LambdaStepMetadata; - - /** - * @public - *

    The configurations and outcomes of an Amazon EMR step execution.

    - */ - EMR?: EMRStepMetadata; - - /** - * @public - *

    The configurations and outcomes of the check step execution. This includes:

    - *
      - *
    • - *

      The type of the check conducted.

      - *
      • - *
    • - *

      The Amazon S3 URIs of baseline constraints and statistics files to be used for the drift check.

      - *
      • - *
    • - *

      The Amazon S3 URIs of newly calculated baseline constraints and statistics.

      - *
      • - *
    • - *

      The model package group name provided.

      - *
      • - *
    • - *

      The Amazon S3 URI of the violation report if violations detected.

      - *
      • - *
    • - *

      The Amazon Resource Name (ARN) of check processing job initiated by the step execution.

      - *
      • - *
    • - *

      The Boolean flags indicating if the drift check is skipped.

      - *
      • - *
    • - *

      If step property BaselineUsedForDriftCheck is set the same as - * CalculatedBaseline.

      - *
      • - *
    - */ - QualityCheck?: QualityCheckStepMetadata; - - /** - * @public - *

    Container for the metadata for a Clarify check step. The configurations - * and outcomes of the check step execution. This includes:

    - *
      - *
    • - *

      The type of the check conducted,

      - *
      • - *
    • - *

      The Amazon S3 URIs of baseline constraints and statistics files to be used for the drift check.

      - *
      • - *
    • - *

      The Amazon S3 URIs of newly calculated baseline constraints and statistics.

      - *
      • - *
    • - *

      The model package group name provided.

      - *
      • - *
    • - *

      The Amazon S3 URI of the violation report if violations detected.

      - *
      • - *
    • - *

      The Amazon Resource Name (ARN) of check processing job initiated by the step execution.

      - *
      • - *
    • - *

      The boolean flags indicating if the drift check is skipped.

      - *
      • - *
    • - *

      If step property BaselineUsedForDriftCheck is set the same as - * CalculatedBaseline.

      - *
      • - *
    - */ - ClarifyCheck?: ClarifyCheckStepMetadata; - - /** - * @public - *

    The configurations and outcomes of a Fail step execution.

    - */ - Fail?: FailStepMetadata; - - /** - * @public - *

    The Amazon Resource Name (ARN) of the AutoML job that was run by this step.

    - */ - AutoMLJob?: AutoMLJobStepMetadata; -} diff --git a/clients/client-sagemaker/src/models/models_4.ts b/clients/client-sagemaker/src/models/models_4.ts index b56079dba9d6b..ff24555f801c8 100644 --- a/clients/client-sagemaker/src/models/models_4.ts +++ b/clients/client-sagemaker/src/models/models_4.ts @@ -9,14 +9,18 @@ import { AppSecurityGroupManagement, AppSpecification, AppType, + AutoMLJobStepMetadata, BatchDataCaptureConfig, BatchStrategy, BatchTransformInput, BooleanOperator, CacheHitResult, + CallbackStepMetadata, Channel, CheckpointConfig, + ClarifyCheckStepMetadata, ClusterInstanceGroupSpecification, + ConditionStepMetadata, ContainerDefinition, InferenceSpecification, JupyterLabAppImageConfig, @@ -97,7 +101,6 @@ import { FeatureParameter, HyperParameterTrainingJobSummary, MemberDefinition, - MetricData, ModelArtifacts, ModelClientConfig, ModelPackageGroupStatus, @@ -133,26 +136,36 @@ import { Direction, DomainSettingsForUpdate, Edge, + EMRStepMetadata, Endpoint, Experiment, + FailStepMetadata, FeatureGroup, FeatureMetadata, Filter, GitConfigForUpdate, HyperParameterTuningJobSearchEntity, InferenceExperimentStopDesiredState, + LambdaStepMetadata, LineageType, + MetricData, + ModelStepMetadata, MonitoringAlertSummary, - PipelineExecutionStepMetadata, + ProcessingJobStepMetadata, + QualityCheckStepMetadata, + RegisterModelStepMetadata, ResourceType, SecondaryStatus, SecondaryStatusTransition, SortBy, SortOrder, + TrainingJobStepMetadata, TransformJobStatus, + TransformJobStepMetadata, TrialComponentMetricSummary, TrialComponentSource, TrialSource, + TuningJobStepMetaData, UserProfileStatus, WarmPoolResourceStatus, WarmPoolStatus, @@ -160,6 +173,153 @@ import { Workteam, } from "./models_3"; +/** + * @public + *

    Metadata for a step execution.

    + */ +export interface PipelineExecutionStepMetadata { + /** + * @public + *

    The Amazon Resource Name (ARN) of the training job that was run by this step execution.

    + */ + TrainingJob?: TrainingJobStepMetadata; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the processing job that was run by this step execution.

    + */ + ProcessingJob?: ProcessingJobStepMetadata; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the transform job that was run by this step execution.

    + */ + TransformJob?: TransformJobStepMetadata; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the tuning job that was run by this step execution.

    + */ + TuningJob?: TuningJobStepMetaData; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the model that was created by this step execution.

    + */ + Model?: ModelStepMetadata; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the model package that the model was registered to by this step execution.

    + */ + RegisterModel?: RegisterModelStepMetadata; + + /** + * @public + *

    The outcome of the condition evaluation that was run by this step execution.

    + */ + Condition?: ConditionStepMetadata; + + /** + * @public + *

    The URL of the Amazon SQS queue used by this step execution, the pipeline generated token, + * and a list of output parameters.

    + */ + Callback?: CallbackStepMetadata; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the Lambda function that was run by this step execution and a list of + * output parameters.

    + */ + Lambda?: LambdaStepMetadata; + + /** + * @public + *

    The configurations and outcomes of an Amazon EMR step execution.

    + */ + EMR?: EMRStepMetadata; + + /** + * @public + *

    The configurations and outcomes of the check step execution. This includes:

    + *
      + *
    • + *

      The type of the check conducted.

      + *
      • + *
    • + *

      The Amazon S3 URIs of baseline constraints and statistics files to be used for the drift check.

      + *
      • + *
    • + *

      The Amazon S3 URIs of newly calculated baseline constraints and statistics.

      + *
      • + *
    • + *

      The model package group name provided.

      + *
      • + *
    • + *

      The Amazon S3 URI of the violation report if violations detected.

      + *
      • + *
    • + *

      The Amazon Resource Name (ARN) of check processing job initiated by the step execution.

      + *
      • + *
    • + *

      The Boolean flags indicating if the drift check is skipped.

      + *
      • + *
    • + *

      If step property BaselineUsedForDriftCheck is set the same as + * CalculatedBaseline.

      + *
      • + *
    + */ + QualityCheck?: QualityCheckStepMetadata; + + /** + * @public + *

    Container for the metadata for a Clarify check step. The configurations + * and outcomes of the check step execution. This includes:

    + *
      + *
    • + *

      The type of the check conducted,

      + *
      • + *
    • + *

      The Amazon S3 URIs of baseline constraints and statistics files to be used for the drift check.

      + *
      • + *
    • + *

      The Amazon S3 URIs of newly calculated baseline constraints and statistics.

      + *
      • + *
    • + *

      The model package group name provided.

      + *
      • + *
    • + *

      The Amazon S3 URI of the violation report if violations detected.

      + *
      • + *
    • + *

      The Amazon Resource Name (ARN) of check processing job initiated by the step execution.

      + *
      • + *
    • + *

      The boolean flags indicating if the drift check is skipped.

      + *
      • + *
    • + *

      If step property BaselineUsedForDriftCheck is set the same as + * CalculatedBaseline.

      + *
      • + *
    + */ + ClarifyCheck?: ClarifyCheckStepMetadata; + + /** + * @public + *

    The configurations and outcomes of a Fail step execution.

    + */ + Fail?: FailStepMetadata; + + /** + * @public + *

    The Amazon Resource Name (ARN) of the AutoML job that was run by this step.

    + */ + AutoMLJob?: AutoMLJobStepMetadata; +} + /** * @public *

    The ARN from an execution of the current pipeline.

    @@ -3987,6 +4147,20 @@ export interface RegisterDevicesRequest { Tags?: Tag[]; } +/** + * @public + *

    Configuration for remote debugging for the UpdateTrainingJob API. To learn more about the remote + * debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) + * for remote debugging.

    + */ +export interface RemoteDebugConfigForUpdate { + /** + * @public + *

    If set to True, enables remote debugging.

    + */ + EnableRemoteDebug?: boolean; +} + /** * @public *

    Contains input values for a task.

    @@ -4136,6 +4310,24 @@ export const SearchSortOrder = { */ export type SearchSortOrder = (typeof SearchSortOrder)[keyof typeof SearchSortOrder]; +/** + * @public + *

    The list of key-value pairs that you specify for your resources.

    + */ +export interface VisibilityConditions { + /** + * @public + *

    The key for that specifies the tag that you're using to filter the search results. The key must start with Tags..

    + */ + Key?: string; + + /** + * @public + *

    The value for the tag that you're using to filter the search results.

    + */ + Value?: string; +} + /** * @public *

    Contains information about a training job.

    @@ -6938,6 +7130,16 @@ export interface UpdateTrainingJobRequest { * length.

    */ ResourceConfig?: ResourceConfigForUpdate; + + /** + * @public + *

    Configuration for remote debugging while the training job is running. You can + * update the remote debugging configuration when the SecondaryStatus of the job + * is Downloading or Training.To learn more about the remote + * debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) + * for remote debugging.

    + */ + RemoteDebugConfig?: RemoteDebugConfigForUpdate; } /** @@ -7324,6 +7526,14 @@ export interface SearchRequest { *

    */ CrossAccountFilterOption?: CrossAccountFilterOption; + + /** + * @public + *

    + * Limits the results of your search request to the resources that you can access. + *

    + */ + VisibilityConditions?: VisibilityConditions[]; } /** diff --git a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts index 79b907565fa43..6b775c3b00877 100644 --- a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts +++ b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts @@ -1528,7 +1528,6 @@ import { LabelingJobOutput, LastUpdateStatus, MemberDefinition, - MetricData, ModelArtifacts, ModelCardExportArtifacts, ModelClientConfig, @@ -1552,6 +1551,7 @@ import { ProfilerRuleConfiguration, RealTimeInferenceRecommendation, RecommendationMetrics, + RemoteDebugConfig, ResolvedAttributes, RetentionPolicy, SelectedStep, @@ -1758,6 +1758,7 @@ import { ListPipelineExecutionsRequest, ListPipelineExecutionsResponse, ListPipelineExecutionStepsRequest, + MetricData, MetricSpecification, ModelCardExportJobSummary, ModelCardSummary, @@ -1779,7 +1780,6 @@ import { NotebookInstanceLifecycleConfigSummary, NotebookInstanceSummary, OidcConfigForResponse, - PipelineExecutionStepMetadata, PipelineExecutionSummary, PredefinedMetricSpecification, ProcessingJobStepMetadata, @@ -1863,6 +1863,7 @@ import { Pipeline, PipelineExecution, PipelineExecutionStep, + PipelineExecutionStepMetadata, PipelineSummary, ProcessingJob, ProcessingJobSummary, @@ -1875,6 +1876,7 @@ import { QueryLineageRequest, QueryLineageResponse, RegisterDevicesRequest, + RemoteDebugConfigForUpdate, RenderableTask, RenderingError, RenderUiTemplateRequest, @@ -2002,6 +2004,7 @@ import { UserProfileDetails, VariantProperty, Vertex, + VisibilityConditions, } from "../models/models_4"; import { SageMakerServiceException as __BaseException } from "../models/SageMakerServiceException"; @@ -23114,6 +23117,10 @@ const se_QueryLineageRequest = (input: QueryLineageRequest, context: __SerdeCont // se_RegisterDevicesRequest omitted. +// se_RemoteDebugConfig omitted. + +// se_RemoteDebugConfigForUpdate omitted. + // se_RenderableTask omitted. // se_RenderUiTemplateRequest omitted. @@ -23204,6 +23211,7 @@ const se_SearchRequest = (input: SearchRequest, context: __SerdeContext): any => SearchExpression: (_) => se_SearchExpression(_, context), SortBy: [], SortOrder: [], + VisibilityConditions: _json, }); }; @@ -23597,6 +23605,10 @@ const se_UpdateTrialComponentRequest = (input: UpdateTrialComponentRequest, cont // se_VectorConfig omitted. +// se_VisibilityConditions omitted. + +// se_VisibilityConditionsList omitted. + // se_VpcConfig omitted. // se_VpcSecurityGroupIds omitted. @@ -27791,6 +27803,7 @@ const de_DescribeTrainingJobResponse = (output: any, context: __SerdeContext): D ProfilerRuleConfigurations: (_: any) => de_ProfilerRuleConfigurations(_, context), ProfilerRuleEvaluationStatuses: (_: any) => de_ProfilerRuleEvaluationStatuses(_, context), ProfilingStatus: __expectString, + RemoteDebugConfig: (_: any) => de_RemoteDebugConfig(_, context), ResourceConfig: (_: any) => de_ResourceConfig(_, context), RetryStrategy: (_: any) => de_RetryStrategy(_, context), RoleArn: __expectString, @@ -34256,6 +34269,15 @@ const de_RegisterModelStepMetadata = (output: any, context: __SerdeContext): Reg }) as any; }; +/** + * deserializeAws_json1_1RemoteDebugConfig + */ +const de_RemoteDebugConfig = (output: any, context: __SerdeContext): RemoteDebugConfig => { + return take(output, { + EnableRemoteDebug: __expectBoolean, + }) as any; +}; + /** * deserializeAws_json1_1RenderingError */ diff --git a/codegen/sdk-codegen/aws-models/sagemaker.json b/codegen/sdk-codegen/aws-models/sagemaker.json index 21b1282be46c0..3a1f39c837c6e 100644 --- a/codegen/sdk-codegen/aws-models/sagemaker.json +++ b/codegen/sdk-codegen/aws-models/sagemaker.json @@ -12226,6 +12226,12 @@ "smithy.api#documentation": "

    The number of times to retry the job when the job fails due to an\n InternalServerError.

    " } }, + "RemoteDebugConfig": { + "target": "com.amazonaws.sagemaker#RemoteDebugConfig", + "traits": { + "smithy.api#documentation": "

    Configuration for remote debugging. To learn more about the remote \n debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) \n for remote debugging.

    " + } + }, "InfraCheckConfig": { "target": "com.amazonaws.sagemaker#InfraCheckConfig", "traits": { @@ -22921,6 +22927,12 @@ "smithy.api#documentation": "

    The number of times to retry the job when the job fails due to an\n InternalServerError.

    " } }, + "RemoteDebugConfig": { + "target": "com.amazonaws.sagemaker#RemoteDebugConfig", + "traits": { + "smithy.api#documentation": "

    Configuration for remote debugging. To learn more about the remote \n debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) \n for remote debugging.

    " + } + }, "InfraCheckConfig": { "target": "com.amazonaws.sagemaker#InfraCheckConfig", "traits": { @@ -25375,6 +25387,9 @@ "com.amazonaws.sagemaker#EnableIotRoleAlias": { "type": "boolean" }, + "com.amazonaws.sagemaker#EnableRemoteDebug": { + "type": "boolean" + }, "com.amazonaws.sagemaker#EnableSagemakerServicecatalogPortfolio": { "type": "operation", "input": { @@ -52188,6 +52203,34 @@ "smithy.api#pattern": ".*" } }, + "com.amazonaws.sagemaker#RemoteDebugConfig": { + "type": "structure", + "members": { + "EnableRemoteDebug": { + "target": "com.amazonaws.sagemaker#EnableRemoteDebug", + "traits": { + "smithy.api#documentation": "

    If set to True, enables remote debugging.

    " + } + } + }, + "traits": { + "smithy.api#documentation": "

    Configuration for remote debugging for the CreateTrainingJob API. To learn more about the remote \n debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) \n for remote debugging.

    " + } + }, + "com.amazonaws.sagemaker#RemoteDebugConfigForUpdate": { + "type": "structure", + "members": { + "EnableRemoteDebug": { + "target": "com.amazonaws.sagemaker#EnableRemoteDebug", + "traits": { + "smithy.api#documentation": "

    If set to True, enables remote debugging.

    " + } + } + }, + "traits": { + "smithy.api#documentation": "

    Configuration for remote debugging for the UpdateTrainingJob API. To learn more about the remote \n debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) \n for remote debugging.

    " + } + }, "com.amazonaws.sagemaker#RenderUiTemplate": { "type": "operation", "input": { @@ -55634,6 +55677,12 @@ "traits": { "smithy.api#documentation": "

    \n A cross account filter option. When the value is \"CrossAccount\" the \n search results will only include resources made discoverable to you from other \n accounts. When the value is \"SameAccount\" or null the \n search results will only include resources from your account. Default is \n null. For more information on searching for resources made \n discoverable to your account, see \n Search discoverable resources in the SageMaker Developer Guide.\n The maximum number of ResourceCatalogs viewable is 1000.\n

    " } + }, + "VisibilityConditions": { + "target": "com.amazonaws.sagemaker#VisibilityConditionsList", + "traits": { + "smithy.api#documentation": "

    \n Limits the results of your search request to the resources that you can access.\n

    " + } } }, "traits": { @@ -55818,7 +55867,7 @@ "StatusMessage": { "target": "com.amazonaws.sagemaker#StatusMessage", "traits": { - "smithy.api#documentation": "

    A detailed description of the progress within a secondary status.\n

    \n

    SageMaker provides secondary statuses and status messages that apply to each of\n them:

    \n
    \n
    Starting
    \n
    \n
      \n
    • \n

      Starting the training job.

      \n
      • \n
    • \n

      Launching requested ML\n instances.

      \n
      • \n
    • \n

      Insufficient\n capacity error from EC2 while launching instances,\n retrying!

      \n
      • \n
    • \n

      Launched\n instance was unhealthy, replacing it!

      \n
      • \n
    • \n

      Preparing the instances for training.

      \n
      • \n
    \n
    \n
    Training
    \n
    \n
      \n
    • \n

      Downloading the training image.

      \n
      • \n
    • \n

      Training\n image download completed. Training in\n progress.

      \n
      • \n
    \n
    \n
    \n \n

    Status messages are subject to change. Therefore, we recommend not including them\n in code that programmatically initiates actions. For examples, don't use status\n messages in if statements.

    \n     \n

    To have an overview of your training job's progress, view\n TrainingJobStatus and SecondaryStatus in DescribeTrainingJob, and StatusMessage together. For example,\n at the start of a training job, you might see the following:

    \n
      \n
    • \n

      \n TrainingJobStatus - InProgress

      \n
      • \n
    • \n

      \n SecondaryStatus - Training

      \n
      • \n
    • \n

      \n StatusMessage - Downloading the training image

      \n
      • \n
    " + "smithy.api#documentation": "

    A detailed description of the progress within a secondary status.\n

    \n

    SageMaker provides secondary statuses and status messages that apply to each of\n them:

    \n
    \n
    Starting
    \n
    \n
      \n
    • \n

      Starting the training job.

      \n
      • \n
    • \n

      Launching requested ML\n instances.

      \n
      • \n
    • \n

      Insufficient\n capacity error from EC2 while launching instances,\n retrying!

      \n
      • \n
    • \n

      Launched\n instance was unhealthy, replacing it!

      \n
      • \n
    • \n

      Preparing the instances for training.

      \n
      • \n
    \n
    \n
    Training
    \n
    \n
      \n
    • \n

      Training\n image download completed. Training in\n progress.

      \n
      • \n
    \n
    \n
    \n \n

    Status messages are subject to change. Therefore, we recommend not including them\n in code that programmatically initiates actions. For examples, don't use status\n messages in if statements.

    \n     \n

    To have an overview of your training job's progress, view\n TrainingJobStatus and SecondaryStatus in DescribeTrainingJob, and StatusMessage together. For example,\n at the start of a training job, you might see the following:

    \n
      \n
    • \n

      \n TrainingJobStatus - InProgress

      \n
      • \n
    • \n

      \n SecondaryStatus - Training

      \n
      • \n
    • \n

      \n StatusMessage - Downloading the training image

      \n
      • \n
    " } } }, @@ -64391,6 +64440,12 @@ "traits": { "smithy.api#documentation": "

    The training job ResourceConfig to update warm pool retention\n length.

    " } + }, + "RemoteDebugConfig": { + "target": "com.amazonaws.sagemaker#RemoteDebugConfigForUpdate", + "traits": { + "smithy.api#documentation": "

    Configuration for remote debugging while the training job is running. You can\n update the remote debugging configuration when the SecondaryStatus of the job\n is Downloading or Training.To learn more about the remote \n debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) \n for remote debugging.

    " + } } }, "traits": { @@ -65299,6 +65354,58 @@ "target": "com.amazonaws.sagemaker#Vertex" } }, + "com.amazonaws.sagemaker#VisibilityConditions": { + "type": "structure", + "members": { + "Key": { + "target": "com.amazonaws.sagemaker#VisibilityConditionsKey", + "traits": { + "smithy.api#documentation": "

    The key for that specifies the tag that you're using to filter the search results. The key must start with Tags..

    " + } + }, + "Value": { + "target": "com.amazonaws.sagemaker#VisibilityConditionsValue", + "traits": { + "smithy.api#documentation": "

    The value for the tag that you're using to filter the search results.

    " + } + } + }, + "traits": { + "smithy.api#documentation": "

    The list of key-value pairs that you specify for your resources.

    " + } + }, + "com.amazonaws.sagemaker#VisibilityConditionsKey": { + "type": "string", + "traits": { + "smithy.api#length": { + "min": 1, + "max": 128 + }, + "smithy.api#pattern": "^([\\p{L}\\p{Z}\\p{N}_.:/=+\\-@]*)$" + } + }, + "com.amazonaws.sagemaker#VisibilityConditionsList": { + "type": "list", + "member": { + "target": "com.amazonaws.sagemaker#VisibilityConditions" + }, + "traits": { + "smithy.api#length": { + "min": 1, + "max": 5 + } + } + }, + "com.amazonaws.sagemaker#VisibilityConditionsValue": { + "type": "string", + "traits": { + "smithy.api#length": { + "min": 0, + "max": 256 + }, + "smithy.api#pattern": "^([\\p{L}\\p{Z}\\p{N}_.:/=+\\-@]*)$" + } + }, "com.amazonaws.sagemaker#VolumeSizeInGB": { "type": "integer", "traits": {