@aws-sdk/client-sagemaker 3.1054.0 → 3.1056.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (40) hide show
  1. package/dist-cjs/index.js +6 -0
  2. package/dist-cjs/schemas/schemas_0.js +80 -49
  3. package/dist-es/models/enums.js +5 -0
  4. package/dist-es/schemas/schemas_0.js +44 -13
  5. package/dist-types/commands/CreateAIRecommendationJobCommand.d.ts +1 -1
  6. package/dist-types/commands/CreateClusterCommand.d.ts +9 -0
  7. package/dist-types/commands/CreateOptimizationJobCommand.d.ts +1 -2
  8. package/dist-types/commands/DescribeAIRecommendationJobCommand.d.ts +3 -3
  9. package/dist-types/commands/DescribeClusterCommand.d.ts +14 -0
  10. package/dist-types/commands/DescribeTrainingPlanCommand.d.ts +1 -0
  11. package/dist-types/commands/ListModelBiasJobDefinitionsCommand.d.ts +1 -1
  12. package/dist-types/commands/ListModelCardExportJobsCommand.d.ts +1 -2
  13. package/dist-types/commands/ListTrainingPlansCommand.d.ts +1 -0
  14. package/dist-types/commands/UpdateClusterCommand.d.ts +9 -0
  15. package/dist-types/commands/UpdateTrialCommand.d.ts +2 -1
  16. package/dist-types/commands/UpdateTrialComponentCommand.d.ts +1 -1
  17. package/dist-types/commands/UpdateUserProfileCommand.d.ts +1 -2
  18. package/dist-types/models/enums.d.ts +13 -0
  19. package/dist-types/models/models_0.d.ts +65 -100
  20. package/dist-types/models/models_1.d.ts +106 -174
  21. package/dist-types/models/models_2.d.ts +180 -102
  22. package/dist-types/models/models_3.d.ts +108 -160
  23. package/dist-types/models/models_4.d.ts +165 -104
  24. package/dist-types/models/models_5.d.ts +102 -1
  25. package/dist-types/schemas/schemas_0.d.ts +4 -0
  26. package/dist-types/ts3.4/commands/CreateOptimizationJobCommand.d.ts +4 -2
  27. package/dist-types/ts3.4/commands/ListModelBiasJobDefinitionsCommand.d.ts +1 -1
  28. package/dist-types/ts3.4/commands/ListModelCardExportJobsCommand.d.ts +4 -2
  29. package/dist-types/ts3.4/commands/UpdateTrialCommand.d.ts +2 -1
  30. package/dist-types/ts3.4/commands/UpdateTrialComponentCommand.d.ts +1 -1
  31. package/dist-types/ts3.4/commands/UpdateUserProfileCommand.d.ts +4 -2
  32. package/dist-types/ts3.4/models/enums.d.ts +7 -0
  33. package/dist-types/ts3.4/models/models_0.d.ts +17 -26
  34. package/dist-types/ts3.4/models/models_1.d.ts +30 -76
  35. package/dist-types/ts3.4/models/models_2.d.ts +85 -28
  36. package/dist-types/ts3.4/models/models_3.d.ts +35 -43
  37. package/dist-types/ts3.4/models/models_4.d.ts +43 -26
  38. package/dist-types/ts3.4/models/models_5.d.ts +28 -0
  39. package/dist-types/ts3.4/schemas/schemas_0.d.ts +4 -0
  40. package/package.json +7 -7
@@ -1,7 +1,108 @@
1
1
  import type { BooleanOperator, CrossAccountFilterOption, ResourceType, SearchSortOrder, WorkforceIpAddressType } from "./enums";
2
- import type { MemberDefinition, NotificationConfiguration, OidcConfig, SourceIpConfig, WorkerAccessConfiguration, WorkforceVpcConfigRequest } from "./models_2";
2
+ import type { UserSettings } from "./models_1";
3
+ import type { MemberDefinition, NotificationConfiguration, OidcConfig, SourceIpConfig, TrialComponentArtifact, TrialComponentParameterValue, TrialComponentStatus, WorkerAccessConfiguration, WorkforceVpcConfigRequest } from "./models_2";
3
4
  import type { Filter, Workforce, Workteam } from "./models_3";
4
5
  import type { NestedFilters, VisibilityConditions } from "./models_4";
6
+ /**
7
+ * @public
8
+ */
9
+ export interface UpdateTrialResponse {
10
+ /**
11
+ * <p>The Amazon Resource Name (ARN) of the trial.</p>
12
+ * @public
13
+ */
14
+ TrialArn?: string | undefined;
15
+ }
16
+ /**
17
+ * @public
18
+ */
19
+ export interface UpdateTrialComponentRequest {
20
+ /**
21
+ * <p>The name of the component to update.</p>
22
+ * @public
23
+ */
24
+ TrialComponentName: string | undefined;
25
+ /**
26
+ * <p>The name of the component as displayed. The name doesn't need to be unique. If <code>DisplayName</code> isn't specified, <code>TrialComponentName</code> is displayed.</p>
27
+ * @public
28
+ */
29
+ DisplayName?: string | undefined;
30
+ /**
31
+ * <p>The new status of the component.</p>
32
+ * @public
33
+ */
34
+ Status?: TrialComponentStatus | undefined;
35
+ /**
36
+ * <p>When the component started.</p>
37
+ * @public
38
+ */
39
+ StartTime?: Date | undefined;
40
+ /**
41
+ * <p>When the component ended.</p>
42
+ * @public
43
+ */
44
+ EndTime?: Date | undefined;
45
+ /**
46
+ * <p>Replaces all of the component's hyperparameters with the specified hyperparameters or add new hyperparameters. Existing hyperparameters are replaced if the trial component is updated with an identical hyperparameter key.</p>
47
+ * @public
48
+ */
49
+ Parameters?: Record<string, TrialComponentParameterValue> | undefined;
50
+ /**
51
+ * <p>The hyperparameters to remove from the component.</p>
52
+ * @public
53
+ */
54
+ ParametersToRemove?: string[] | undefined;
55
+ /**
56
+ * <p>Replaces all of the component's input artifacts with the specified artifacts or adds new input artifacts. Existing input artifacts are replaced if the trial component is updated with an identical input artifact key.</p>
57
+ * @public
58
+ */
59
+ InputArtifacts?: Record<string, TrialComponentArtifact> | undefined;
60
+ /**
61
+ * <p>The input artifacts to remove from the component.</p>
62
+ * @public
63
+ */
64
+ InputArtifactsToRemove?: string[] | undefined;
65
+ /**
66
+ * <p>Replaces all of the component's output artifacts with the specified artifacts or adds new output artifacts. Existing output artifacts are replaced if the trial component is updated with an identical output artifact key.</p>
67
+ * @public
68
+ */
69
+ OutputArtifacts?: Record<string, TrialComponentArtifact> | undefined;
70
+ /**
71
+ * <p>The output artifacts to remove from the component.</p>
72
+ * @public
73
+ */
74
+ OutputArtifactsToRemove?: string[] | undefined;
75
+ }
76
+ /**
77
+ * @public
78
+ */
79
+ export interface UpdateTrialComponentResponse {
80
+ /**
81
+ * <p>The Amazon Resource Name (ARN) of the trial component.</p>
82
+ * @public
83
+ */
84
+ TrialComponentArn?: string | undefined;
85
+ }
86
+ /**
87
+ * @public
88
+ */
89
+ export interface UpdateUserProfileRequest {
90
+ /**
91
+ * <p>The domain ID.</p>
92
+ * @public
93
+ */
94
+ DomainId: string | undefined;
95
+ /**
96
+ * <p>The user profile name.</p>
97
+ * @public
98
+ */
99
+ UserProfileName: string | undefined;
100
+ /**
101
+ * <p>A collection of settings.</p>
102
+ * @public
103
+ */
104
+ UserSettings?: UserSettings | undefined;
105
+ }
5
106
  /**
6
107
  * @public
7
108
  */
@@ -195,8 +195,12 @@ export declare var ClusterOrchestrator$: StaticStructureSchema;
195
195
  export declare var ClusterOrchestratorEksConfig$: StaticStructureSchema;
196
196
  export declare var ClusterOrchestratorSlurmConfig$: StaticStructureSchema;
197
197
  export declare var ClusterRestrictedInstanceGroupDetails$: StaticStructureSchema;
198
+ export declare var ClusterRestrictedInstanceGroupsConfig$: StaticStructureSchema;
199
+ export declare var ClusterRestrictedInstanceGroupsConfigOutput$: StaticStructureSchema;
198
200
  export declare var ClusterRestrictedInstanceGroupSpecification$: StaticStructureSchema;
199
201
  export declare var ClusterSchedulerConfigSummary$: StaticStructureSchema;
202
+ export declare var ClusterSharedEnvironmentConfig$: StaticStructureSchema;
203
+ export declare var ClusterSharedEnvironmentConfigDetails$: StaticStructureSchema;
200
204
  export declare var ClusterSlurmConfig$: StaticStructureSchema;
201
205
  export declare var ClusterSlurmConfigDetails$: StaticStructureSchema;
202
206
  export declare var ClusterSpotOptions$: StaticStructureSchema;
@@ -1,7 +1,9 @@
1
1
  import { Command as $Command } from "@smithy/core/client";
2
2
  import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
- import { CreateOptimizationJobRequest } from "../models/models_1";
4
- import { CreateOptimizationJobResponse } from "../models/models_2";
3
+ import {
4
+ CreateOptimizationJobRequest,
5
+ CreateOptimizationJobResponse,
6
+ } from "../models/models_2";
5
7
  import {
6
8
  SageMakerClientResolvedConfig,
7
9
  ServiceInputTypes,
@@ -3,7 +3,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
3
  import {
4
4
  ListModelBiasJobDefinitionsRequest,
5
5
  ListModelBiasJobDefinitionsResponse,
6
- } from "../models/models_3";
6
+ } from "../models/models_4";
7
7
  import {
8
8
  SageMakerClientResolvedConfig,
9
9
  ServiceInputTypes,
@@ -1,7 +1,9 @@
1
1
  import { Command as $Command } from "@smithy/core/client";
2
2
  import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
- import { ListModelCardExportJobsRequest } from "../models/models_3";
4
- import { ListModelCardExportJobsResponse } from "../models/models_4";
3
+ import {
4
+ ListModelCardExportJobsRequest,
5
+ ListModelCardExportJobsResponse,
6
+ } from "../models/models_4";
5
7
  import {
6
8
  SageMakerClientResolvedConfig,
7
9
  ServiceInputTypes,
@@ -1,6 +1,7 @@
1
1
  import { Command as $Command } from "@smithy/core/client";
2
2
  import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
- import { UpdateTrialRequest, UpdateTrialResponse } from "../models/models_4";
3
+ import { UpdateTrialRequest } from "../models/models_4";
4
+ import { UpdateTrialResponse } from "../models/models_5";
4
5
  import {
5
6
  SageMakerClientResolvedConfig,
6
7
  ServiceInputTypes,
@@ -3,7 +3,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
3
  import {
4
4
  UpdateTrialComponentRequest,
5
5
  UpdateTrialComponentResponse,
6
- } from "../models/models_4";
6
+ } from "../models/models_5";
7
7
  import {
8
8
  SageMakerClientResolvedConfig,
9
9
  ServiceInputTypes,
@@ -1,7 +1,9 @@
1
1
  import { Command as $Command } from "@smithy/core/client";
2
2
  import { MetadataBearer as __MetadataBearer } from "@smithy/types";
3
- import { UpdateUserProfileRequest } from "../models/models_4";
4
- import { UpdateUserProfileResponse } from "../models/models_5";
3
+ import {
4
+ UpdateUserProfileRequest,
5
+ UpdateUserProfileResponse,
6
+ } from "../models/models_5";
5
7
  import {
6
8
  SageMakerClientResolvedConfig,
7
9
  ServiceInputTypes,
@@ -666,6 +666,7 @@ export declare const AIRecommendationInstanceType: {
666
666
  readonly ML_P5E_48XLARGE: "ml.p5e.48xlarge";
667
667
  readonly ML_P5_48XLARGE: "ml.p5.48xlarge";
668
668
  readonly ML_P5_4XLARGE: "ml.p5.4xlarge";
669
+ readonly ML_P6_B200_48XLARGE: "ml.p6-b200.48xlarge";
669
670
  };
670
671
  export type AIRecommendationInstanceType =
671
672
  (typeof AIRecommendationInstanceType)[keyof typeof AIRecommendationInstanceType];
@@ -1683,6 +1684,12 @@ export declare const ClusterEventResourceType: {
1683
1684
  };
1684
1685
  export type ClusterEventResourceType =
1685
1686
  (typeof ClusterEventResourceType)[keyof typeof ClusterEventResourceType];
1687
+ export declare const ClusterFSxLustreDeletionPolicy: {
1688
+ readonly DELETE_IF_NOT_USED: "DeleteIfNotUsed";
1689
+ readonly KEEP: "Keep";
1690
+ };
1691
+ export type ClusterFSxLustreDeletionPolicy =
1692
+ (typeof ClusterFSxLustreDeletionPolicy)[keyof typeof ClusterFSxLustreDeletionPolicy];
1686
1693
  export declare const ClusterImageVersionStatus: {
1687
1694
  readonly UPDATE_AVAILABLE: "UpdateAvailable";
1688
1695
  readonly UP_TO_DATE: "UpToDate";
@@ -1,6 +1,5 @@
1
1
  import {
2
2
  ActionStatus,
3
- ActivationState,
4
3
  ActiveClusterOperationName,
5
4
  AdditionalS3DataSourceDataType,
6
5
  AggregationTransformationValue,
@@ -52,6 +51,7 @@ import {
52
51
  ClusterConfigMode,
53
52
  ClusterEventLevel,
54
53
  ClusterEventResourceType,
54
+ ClusterFSxLustreDeletionPolicy,
55
55
  ClusterImageVersionStatus,
56
56
  ClusterInstanceStatus,
57
57
  ClusterInstanceType,
@@ -62,7 +62,6 @@ import {
62
62
  ClusterStatus,
63
63
  CompilationJobStatus,
64
64
  CompressionType,
65
- ConditionOutcome,
66
65
  DataSourceName,
67
66
  DeepHealthCheckType,
68
67
  DetailedAlgorithmStatus,
@@ -87,7 +86,6 @@ import {
87
86
  ProcessingS3InputMode,
88
87
  ProductionVariantInstanceType,
89
88
  RecordWrapper,
90
- RepositoryAccessMode,
91
89
  ResourceSharingStrategy,
92
90
  S3DataDistribution,
93
91
  S3DataType,
@@ -1800,6 +1798,22 @@ export interface ClusterRestrictedInstanceGroupDetails {
1800
1798
  ScheduledUpdateConfig?: ScheduledUpdateConfig | undefined;
1801
1799
  EnvironmentConfig?: EnvironmentConfigDetails | undefined;
1802
1800
  }
1801
+ export interface ClusterSharedEnvironmentConfig {
1802
+ FSxLustreDeletionPolicy: ClusterFSxLustreDeletionPolicy | undefined;
1803
+ FSxLustreConfig: FSxLustreConfig | undefined;
1804
+ }
1805
+ export interface ClusterRestrictedInstanceGroupsConfig {
1806
+ SharedEnvironmentConfig: ClusterSharedEnvironmentConfig | undefined;
1807
+ }
1808
+ export interface ClusterSharedEnvironmentConfigDetails {
1809
+ CurrentFSxLustreConfig?: FSxLustreConfig | undefined;
1810
+ DesiredFSxLustreConfig?: FSxLustreConfig | undefined;
1811
+ CurrentFSxLustreDeletionPolicy?: ClusterFSxLustreDeletionPolicy | undefined;
1812
+ DesiredFSxLustreDeletionPolicy?: ClusterFSxLustreDeletionPolicy | undefined;
1813
+ }
1814
+ export interface ClusterRestrictedInstanceGroupsConfigOutput {
1815
+ SharedEnvironmentConfig: ClusterSharedEnvironmentConfigDetails | undefined;
1816
+ }
1803
1817
  export interface EnvironmentConfig {
1804
1818
  FSxLustreConfig?: FSxLustreConfig | undefined;
1805
1819
  }
@@ -1924,26 +1938,3 @@ export interface ComputeQuotaTarget {
1924
1938
  TeamName: string | undefined;
1925
1939
  FairShareWeight?: number | undefined;
1926
1940
  }
1927
- export interface ComputeQuotaSummary {
1928
- ComputeQuotaArn: string | undefined;
1929
- ComputeQuotaId: string | undefined;
1930
- Name: string | undefined;
1931
- ComputeQuotaVersion?: number | undefined;
1932
- Status: SchedulerResourceStatus | undefined;
1933
- ClusterArn?: string | undefined;
1934
- ComputeQuotaConfig?: ComputeQuotaConfig | undefined;
1935
- ComputeQuotaTarget: ComputeQuotaTarget | undefined;
1936
- ActivationState?: ActivationState | undefined;
1937
- CreationTime: Date | undefined;
1938
- LastModifiedTime?: Date | undefined;
1939
- }
1940
- export interface ConditionStepMetadata {
1941
- Outcome?: ConditionOutcome | undefined;
1942
- }
1943
- export interface RepositoryAuthConfig {
1944
- RepositoryCredentialsProviderArn: string | undefined;
1945
- }
1946
- export interface ImageConfig {
1947
- RepositoryAccessMode: RepositoryAccessMode | undefined;
1948
- RepositoryAuthConfig?: RepositoryAuthConfig | undefined;
1949
- }
@@ -17,6 +17,7 @@ import {
17
17
  ClusterNodeRecovery,
18
18
  CollectionType,
19
19
  CompleteOnConvergence,
20
+ ConditionOutcome,
20
21
  ContainerMode,
21
22
  ContentClassifier,
22
23
  DeviceSubsetType,
@@ -60,7 +61,6 @@ import {
60
61
  MonitoringType,
61
62
  NotebookInstanceAcceleratorType,
62
63
  NotebookOutputOption,
63
- OptimizationJobDeploymentInstanceType,
64
64
  ParameterType,
65
65
  ProblemType,
66
66
  ProcessingInstanceType,
@@ -73,11 +73,13 @@ import {
73
73
  ProductionVariantInstanceType,
74
74
  RecommendationJobSupportedEndpointType,
75
75
  RecommendationJobType,
76
+ RepositoryAccessMode,
76
77
  RootAccess,
77
78
  RoutingStrategy,
78
79
  RStudioServerProAccessStatus,
79
80
  RStudioServerProUserGroup,
80
81
  SageMakerImageName,
82
+ SchedulerResourceStatus,
81
83
  SkipModelValidation,
82
84
  StorageType,
83
85
  StudioWebPortal,
@@ -148,6 +150,7 @@ import {
148
150
  ClusterAutoScalingConfig,
149
151
  ClusterInstanceGroupSpecification,
150
152
  ClusterOrchestrator,
153
+ ClusterRestrictedInstanceGroupsConfig,
151
154
  ClusterRestrictedInstanceGroupSpecification,
152
155
  ClusterTieredStorageConfig,
153
156
  CodeEditorAppImageConfig,
@@ -158,7 +161,6 @@ import {
158
161
  ComputeQuotaTarget,
159
162
  CustomImage,
160
163
  GitConfig,
161
- ImageConfig,
162
164
  InferenceSpecification,
163
165
  JupyterLabAppImageConfig,
164
166
  KernelGatewayImageConfig,
@@ -173,6 +175,29 @@ import {
173
175
  TransformJobDefinition,
174
176
  VpcConfig,
175
177
  } from "./models_0";
178
+ export interface ComputeQuotaSummary {
179
+ ComputeQuotaArn: string | undefined;
180
+ ComputeQuotaId: string | undefined;
181
+ Name: string | undefined;
182
+ ComputeQuotaVersion?: number | undefined;
183
+ Status: SchedulerResourceStatus | undefined;
184
+ ClusterArn?: string | undefined;
185
+ ComputeQuotaConfig?: ComputeQuotaConfig | undefined;
186
+ ComputeQuotaTarget: ComputeQuotaTarget | undefined;
187
+ ActivationState?: ActivationState | undefined;
188
+ CreationTime: Date | undefined;
189
+ LastModifiedTime?: Date | undefined;
190
+ }
191
+ export interface ConditionStepMetadata {
192
+ Outcome?: ConditionOutcome | undefined;
193
+ }
194
+ export interface RepositoryAuthConfig {
195
+ RepositoryCredentialsProviderArn: string | undefined;
196
+ }
197
+ export interface ImageConfig {
198
+ RepositoryAccessMode: RepositoryAccessMode | undefined;
199
+ RepositoryAuthConfig?: RepositoryAuthConfig | undefined;
200
+ }
176
201
  export interface MultiModelConfig {
177
202
  ModelCacheSetting?: ModelCacheSetting | undefined;
178
203
  }
@@ -398,6 +423,9 @@ export interface CreateClusterRequest {
398
423
  RestrictedInstanceGroups?:
399
424
  | ClusterRestrictedInstanceGroupSpecification[]
400
425
  | undefined;
426
+ RestrictedInstanceGroupsConfig?:
427
+ | ClusterRestrictedInstanceGroupsConfig
428
+ | undefined;
401
429
  VpcConfig?: VpcConfig | undefined;
402
430
  Tags?: Tag[] | undefined;
403
431
  Orchestrator?: ClusterOrchestrator | undefined;
@@ -1905,77 +1933,3 @@ export interface ModelSpeculativeDecodingConfig {
1905
1933
  Technique: ModelSpeculativeDecodingTechnique | undefined;
1906
1934
  TrainingDataSource?: ModelSpeculativeDecodingTrainingDataSource | undefined;
1907
1935
  }
1908
- export type OptimizationConfig =
1909
- | OptimizationConfig.ModelCompilationConfigMember
1910
- | OptimizationConfig.ModelQuantizationConfigMember
1911
- | OptimizationConfig.ModelShardingConfigMember
1912
- | OptimizationConfig.ModelSpeculativeDecodingConfigMember
1913
- | OptimizationConfig.$UnknownMember;
1914
- export declare namespace OptimizationConfig {
1915
- interface ModelQuantizationConfigMember {
1916
- ModelQuantizationConfig: ModelQuantizationConfig;
1917
- ModelCompilationConfig?: never;
1918
- ModelShardingConfig?: never;
1919
- ModelSpeculativeDecodingConfig?: never;
1920
- $unknown?: never;
1921
- }
1922
- interface ModelCompilationConfigMember {
1923
- ModelQuantizationConfig?: never;
1924
- ModelCompilationConfig: ModelCompilationConfig;
1925
- ModelShardingConfig?: never;
1926
- ModelSpeculativeDecodingConfig?: never;
1927
- $unknown?: never;
1928
- }
1929
- interface ModelShardingConfigMember {
1930
- ModelQuantizationConfig?: never;
1931
- ModelCompilationConfig?: never;
1932
- ModelShardingConfig: ModelShardingConfig;
1933
- ModelSpeculativeDecodingConfig?: never;
1934
- $unknown?: never;
1935
- }
1936
- interface ModelSpeculativeDecodingConfigMember {
1937
- ModelQuantizationConfig?: never;
1938
- ModelCompilationConfig?: never;
1939
- ModelShardingConfig?: never;
1940
- ModelSpeculativeDecodingConfig: ModelSpeculativeDecodingConfig;
1941
- $unknown?: never;
1942
- }
1943
- interface $UnknownMember {
1944
- ModelQuantizationConfig?: never;
1945
- ModelCompilationConfig?: never;
1946
- ModelShardingConfig?: never;
1947
- ModelSpeculativeDecodingConfig?: never;
1948
- $unknown: [string, any];
1949
- }
1950
- interface Visitor<T> {
1951
- ModelQuantizationConfig: (value: ModelQuantizationConfig) => T;
1952
- ModelCompilationConfig: (value: ModelCompilationConfig) => T;
1953
- ModelShardingConfig: (value: ModelShardingConfig) => T;
1954
- ModelSpeculativeDecodingConfig: (
1955
- value: ModelSpeculativeDecodingConfig
1956
- ) => T;
1957
- _: (name: string, value: any) => T;
1958
- }
1959
- }
1960
- export interface OptimizationJobOutputConfig {
1961
- KmsKeyId?: string | undefined;
1962
- S3OutputLocation: string | undefined;
1963
- SageMakerModel?: OptimizationSageMakerModel | undefined;
1964
- }
1965
- export interface OptimizationVpcConfig {
1966
- SecurityGroupIds: string[] | undefined;
1967
- Subnets: string[] | undefined;
1968
- }
1969
- export interface CreateOptimizationJobRequest {
1970
- OptimizationJobName: string | undefined;
1971
- RoleArn: string | undefined;
1972
- ModelSource: OptimizationJobModelSource | undefined;
1973
- DeploymentInstanceType: OptimizationJobDeploymentInstanceType | undefined;
1974
- MaxInstanceCount?: number | undefined;
1975
- OptimizationEnvironment?: Record<string, string> | undefined;
1976
- OptimizationConfigs: OptimizationConfig[] | undefined;
1977
- OutputConfig: OptimizationJobOutputConfig | undefined;
1978
- StoppingCondition: StoppingCondition | undefined;
1979
- Tags?: Tag[] | undefined;
1980
- VpcConfig?: OptimizationVpcConfig | undefined;
1981
- }
@@ -50,6 +50,7 @@ import {
50
50
  LastUpdateStatusValue,
51
51
  ObjectiveStatus,
52
52
  OfflineStoreStatusValue,
53
+ OptimizationJobDeploymentInstanceType,
53
54
  PartnerAppAuthType,
54
55
  PartnerAppType,
55
56
  Peft,
@@ -129,6 +130,7 @@ import {
129
130
  ClusterNodeDetails,
130
131
  ClusterOrchestrator,
131
132
  ClusterRestrictedInstanceGroupDetails,
133
+ ClusterRestrictedInstanceGroupsConfigOutput,
132
134
  ClusterTieredStorageConfig,
133
135
  CodeEditorAppImageConfig,
134
136
  CodeRepository,
@@ -175,15 +177,16 @@ import {
175
177
  HyperParameterTrainingJobDefinition,
176
178
  HyperParameterTuningJobConfig,
177
179
  HyperParameterTuningJobWarmStartConfig,
178
- InferenceComponentComputeResourceRequirements,
179
- InferenceComponentSchedulingConfig,
180
- InferenceComponentStartupParameters,
181
180
  InputConfig,
182
181
  JupyterServerAppSettings,
183
182
  KernelGatewayAppSettings,
184
183
  MetadataProperties,
185
184
  MetricsConfig,
185
+ ModelCompilationConfig,
186
186
  ModelDeployConfig,
187
+ ModelQuantizationConfig,
188
+ ModelShardingConfig,
189
+ ModelSpeculativeDecodingConfig,
187
190
  MonitoringNetworkConfig,
188
191
  MonitoringOutputConfig,
189
192
  MonitoringResources,
@@ -192,6 +195,8 @@ import {
192
195
  NetworkConfig,
193
196
  OfflineStoreConfig,
194
197
  OnlineStoreConfig,
198
+ OptimizationJobModelSource,
199
+ OptimizationSageMakerModel,
195
200
  OutputConfig,
196
201
  ProductionVariant,
197
202
  ProductionVariantManagedInstanceScaling,
@@ -202,6 +207,80 @@ import {
202
207
  TrainingSpecification,
203
208
  UserSettings,
204
209
  } from "./models_1";
210
+ export type OptimizationConfig =
211
+ | OptimizationConfig.ModelCompilationConfigMember
212
+ | OptimizationConfig.ModelQuantizationConfigMember
213
+ | OptimizationConfig.ModelShardingConfigMember
214
+ | OptimizationConfig.ModelSpeculativeDecodingConfigMember
215
+ | OptimizationConfig.$UnknownMember;
216
+ export declare namespace OptimizationConfig {
217
+ interface ModelQuantizationConfigMember {
218
+ ModelQuantizationConfig: ModelQuantizationConfig;
219
+ ModelCompilationConfig?: never;
220
+ ModelShardingConfig?: never;
221
+ ModelSpeculativeDecodingConfig?: never;
222
+ $unknown?: never;
223
+ }
224
+ interface ModelCompilationConfigMember {
225
+ ModelQuantizationConfig?: never;
226
+ ModelCompilationConfig: ModelCompilationConfig;
227
+ ModelShardingConfig?: never;
228
+ ModelSpeculativeDecodingConfig?: never;
229
+ $unknown?: never;
230
+ }
231
+ interface ModelShardingConfigMember {
232
+ ModelQuantizationConfig?: never;
233
+ ModelCompilationConfig?: never;
234
+ ModelShardingConfig: ModelShardingConfig;
235
+ ModelSpeculativeDecodingConfig?: never;
236
+ $unknown?: never;
237
+ }
238
+ interface ModelSpeculativeDecodingConfigMember {
239
+ ModelQuantizationConfig?: never;
240
+ ModelCompilationConfig?: never;
241
+ ModelShardingConfig?: never;
242
+ ModelSpeculativeDecodingConfig: ModelSpeculativeDecodingConfig;
243
+ $unknown?: never;
244
+ }
245
+ interface $UnknownMember {
246
+ ModelQuantizationConfig?: never;
247
+ ModelCompilationConfig?: never;
248
+ ModelShardingConfig?: never;
249
+ ModelSpeculativeDecodingConfig?: never;
250
+ $unknown: [string, any];
251
+ }
252
+ interface Visitor<T> {
253
+ ModelQuantizationConfig: (value: ModelQuantizationConfig) => T;
254
+ ModelCompilationConfig: (value: ModelCompilationConfig) => T;
255
+ ModelShardingConfig: (value: ModelShardingConfig) => T;
256
+ ModelSpeculativeDecodingConfig: (
257
+ value: ModelSpeculativeDecodingConfig
258
+ ) => T;
259
+ _: (name: string, value: any) => T;
260
+ }
261
+ }
262
+ export interface OptimizationJobOutputConfig {
263
+ KmsKeyId?: string | undefined;
264
+ S3OutputLocation: string | undefined;
265
+ SageMakerModel?: OptimizationSageMakerModel | undefined;
266
+ }
267
+ export interface OptimizationVpcConfig {
268
+ SecurityGroupIds: string[] | undefined;
269
+ Subnets: string[] | undefined;
270
+ }
271
+ export interface CreateOptimizationJobRequest {
272
+ OptimizationJobName: string | undefined;
273
+ RoleArn: string | undefined;
274
+ ModelSource: OptimizationJobModelSource | undefined;
275
+ DeploymentInstanceType: OptimizationJobDeploymentInstanceType | undefined;
276
+ MaxInstanceCount?: number | undefined;
277
+ OptimizationEnvironment?: Record<string, string> | undefined;
278
+ OptimizationConfigs: OptimizationConfig[] | undefined;
279
+ OutputConfig: OptimizationJobOutputConfig | undefined;
280
+ StoppingCondition: StoppingCondition | undefined;
281
+ Tags?: Tag[] | undefined;
282
+ VpcConfig?: OptimizationVpcConfig | undefined;
283
+ }
205
284
  export interface CreateOptimizationJobResponse {
206
285
  OptimizationJobArn: string | undefined;
207
286
  }
@@ -1316,6 +1395,9 @@ export interface DescribeClusterResponse {
1316
1395
  RestrictedInstanceGroups?:
1317
1396
  | ClusterRestrictedInstanceGroupDetails[]
1318
1397
  | undefined;
1398
+ RestrictedInstanceGroupsConfig?:
1399
+ | ClusterRestrictedInstanceGroupsConfigOutput
1400
+ | undefined;
1319
1401
  VpcConfig?: VpcConfig | undefined;
1320
1402
  Orchestrator?: ClusterOrchestrator | undefined;
1321
1403
  TieredStorageConfig?: ClusterTieredStorageConfig | undefined;
@@ -1935,28 +2017,3 @@ export interface InferenceComponentPlacementStatus {
1935
2017
  InstanceType: ProductionVariantInstanceType | undefined;
1936
2018
  CurrentCopyCount: number | undefined;
1937
2019
  }
1938
- export interface InferenceComponentRuntimeConfigSummary {
1939
- DesiredCopyCount?: number | undefined;
1940
- CurrentCopyCount?: number | undefined;
1941
- PlacementStatus?: InferenceComponentPlacementStatus[] | undefined;
1942
- }
1943
- export interface InferenceComponentContainerSpecificationSummary {
1944
- DeployedImage?: DeployedImage | undefined;
1945
- ArtifactUrl?: string | undefined;
1946
- Environment?: Record<string, string> | undefined;
1947
- }
1948
- export interface InferenceComponentDataCacheConfigSummary {
1949
- EnableCaching: boolean | undefined;
1950
- }
1951
- export interface InferenceComponentSpecificationSummary {
1952
- InstanceType?: ProductionVariantInstanceType | undefined;
1953
- ModelName?: string | undefined;
1954
- Container?: InferenceComponentContainerSpecificationSummary | undefined;
1955
- StartupParameters?: InferenceComponentStartupParameters | undefined;
1956
- ComputeResourceRequirements?:
1957
- | InferenceComponentComputeResourceRequirements
1958
- | undefined;
1959
- BaseInferenceComponentName?: string | undefined;
1960
- DataCacheConfig?: InferenceComponentDataCacheConfigSummary | undefined;
1961
- SchedulingConfig?: InferenceComponentSchedulingConfig | undefined;
1962
- }