aws-sdk-sagemaker 1.209.0

AWS SDK for Amazon SageMaker Service
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
#[allow(clippy::unnecessary_wraps)]
pub fn de_describe_training_job_http_error(
    _response_status: u16,
    _response_headers: &::aws_smithy_runtime_api::http::Headers,
    _response_body: &[u8],
) -> std::result::Result<
    crate::operation::describe_training_job::DescribeTrainingJobOutput,
    crate::operation::describe_training_job::DescribeTrainingJobError,
> {
    #[allow(unused_mut)]
    let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(_response_status, _response_headers, _response_body)
        .map_err(crate::operation::describe_training_job::DescribeTrainingJobError::unhandled)?;
    generic_builder = ::aws_types::request_id::apply_request_id(generic_builder, _response_headers);
    let generic = generic_builder.build();
    let error_code = match generic.code() {
        Some(code) => code,
        None => return Err(crate::operation::describe_training_job::DescribeTrainingJobError::unhandled(generic)),
    };

    let _error_message = generic.message().map(|msg| msg.to_owned());
    Err(match error_code {
        "ResourceNotFound" => crate::operation::describe_training_job::DescribeTrainingJobError::ResourceNotFound({
            #[allow(unused_mut)]
            let mut tmp = {
                #[allow(unused_mut)]
                let mut output = crate::types::error::builders::ResourceNotFoundBuilder::default();
                output = crate::protocol_serde::shape_resource_not_found::de_resource_not_found_json_err(_response_body, output)
                    .map_err(crate::operation::describe_training_job::DescribeTrainingJobError::unhandled)?;
                let output = output.meta(generic);
                output.build()
            };
            if tmp.message.is_none() {
                tmp.message = _error_message;
            }
            tmp
        }),
        _ => crate::operation::describe_training_job::DescribeTrainingJobError::generic(generic),
    })
}

#[allow(clippy::unnecessary_wraps)]
pub fn de_describe_training_job_http_response(
    _response_status: u16,
    _response_headers: &::aws_smithy_runtime_api::http::Headers,
    _response_body: &[u8],
) -> std::result::Result<
    crate::operation::describe_training_job::DescribeTrainingJobOutput,
    crate::operation::describe_training_job::DescribeTrainingJobError,
> {
    Ok({
        #[allow(unused_mut)]
        let mut output = crate::operation::describe_training_job::builders::DescribeTrainingJobOutputBuilder::default();
        output = crate::protocol_serde::shape_describe_training_job::de_describe_training_job(_response_body, output)
            .map_err(crate::operation::describe_training_job::DescribeTrainingJobError::unhandled)?;
        output._set_request_id(::aws_types::request_id::RequestId::request_id(_response_headers).map(str::to_string));
        crate::serde_util::describe_training_job_output_output_correct_errors(output).build()
    })
}

pub fn ser_describe_training_job_input(
    input: &crate::operation::describe_training_job::DescribeTrainingJobInput,
) -> ::std::result::Result<::aws_smithy_types::body::SdkBody, ::aws_smithy_types::error::operation::SerializationError> {
    let mut out = String::new();
    let mut object = ::aws_smithy_json::serialize::JsonObjectWriter::new(&mut out);
    crate::protocol_serde::shape_describe_training_job_input::ser_describe_training_job_input_input(&mut object, input)?;
    object.finish();
    Ok(::aws_smithy_types::body::SdkBody::from(out))
}

pub(crate) fn de_describe_training_job(
    _value: &[u8],
    mut builder: crate::operation::describe_training_job::builders::DescribeTrainingJobOutputBuilder,
) -> ::std::result::Result<
    crate::operation::describe_training_job::builders::DescribeTrainingJobOutputBuilder,
    ::aws_smithy_json::deserialize::error::DeserializeError,
> {
    let mut tokens_owned = ::aws_smithy_json::deserialize::json_token_iter(crate::protocol_serde::or_empty_doc(_value)).peekable();
    let tokens = &mut tokens_owned;
    #[allow(unused_variables)]
    let depth = 0u32;
    ::aws_smithy_json::deserialize::token::expect_start_object(tokens.next())?;
    loop {
        match tokens.next().transpose()? {
            Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
            Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
                "TrainingJobName" => {
                    builder = builder.set_training_job_name(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "TrainingJobArn" => {
                    builder = builder.set_training_job_arn(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "TuningJobArn" => {
                    builder = builder.set_tuning_job_arn(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "LabelingJobArn" => {
                    builder = builder.set_labeling_job_arn(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "AutoMLJobArn" => {
                    builder = builder.set_auto_ml_job_arn(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "ModelArtifacts" => {
                    builder = builder.set_model_artifacts(crate::protocol_serde::shape_model_artifacts::de_model_artifacts(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "TrainingJobStatus" => {
                    builder = builder.set_training_job_status(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| crate::types::TrainingJobStatus::from(u.as_ref())))
                            .transpose()?,
                    );
                }
                "SecondaryStatus" => {
                    builder = builder.set_secondary_status(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| crate::types::SecondaryStatus::from(u.as_ref())))
                            .transpose()?,
                    );
                }
                "FailureReason" => {
                    builder = builder.set_failure_reason(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "HyperParameters" => {
                    builder = builder.set_hyper_parameters(crate::protocol_serde::shape_hyper_parameters::de_hyper_parameters(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "AlgorithmSpecification" => {
                    builder = builder.set_algorithm_specification(crate::protocol_serde::shape_algorithm_specification::de_algorithm_specification(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "RoleArn" => {
                    builder = builder.set_role_arn(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                "InputDataConfig" => {
                    builder = builder.set_input_data_config(crate::protocol_serde::shape_input_data_config::de_input_data_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "OutputDataConfig" => {
                    builder = builder.set_output_data_config(crate::protocol_serde::shape_output_data_config::de_output_data_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "ResourceConfig" => {
                    builder = builder.set_resource_config(crate::protocol_serde::shape_resource_config::de_resource_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "WarmPoolStatus" => {
                    builder = builder.set_warm_pool_status(crate::protocol_serde::shape_warm_pool_status::de_warm_pool_status(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "VpcConfig" => {
                    builder = builder.set_vpc_config(crate::protocol_serde::shape_vpc_config::de_vpc_config(tokens, _value, depth + 1)?);
                }
                "StoppingCondition" => {
                    builder = builder.set_stopping_condition(crate::protocol_serde::shape_stopping_condition::de_stopping_condition(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "CreationTime" => {
                    builder = builder.set_creation_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(
                        tokens.next(),
                        ::aws_smithy_types::date_time::Format::EpochSeconds,
                    )?);
                }
                "TrainingStartTime" => {
                    builder = builder.set_training_start_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(
                        tokens.next(),
                        ::aws_smithy_types::date_time::Format::EpochSeconds,
                    )?);
                }
                "TrainingEndTime" => {
                    builder = builder.set_training_end_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(
                        tokens.next(),
                        ::aws_smithy_types::date_time::Format::EpochSeconds,
                    )?);
                }
                "LastModifiedTime" => {
                    builder = builder.set_last_modified_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(
                        tokens.next(),
                        ::aws_smithy_types::date_time::Format::EpochSeconds,
                    )?);
                }
                "SecondaryStatusTransitions" => {
                    builder = builder.set_secondary_status_transitions(
                        crate::protocol_serde::shape_secondary_status_transitions::de_secondary_status_transitions(tokens, _value, depth + 1)?,
                    );
                }
                "FinalMetricDataList" => {
                    builder = builder.set_final_metric_data_list(crate::protocol_serde::shape_final_metric_data_list::de_final_metric_data_list(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "EnableNetworkIsolation" => {
                    builder = builder.set_enable_network_isolation(::aws_smithy_json::deserialize::token::expect_bool_or_null(tokens.next())?);
                }
                "EnableInterContainerTrafficEncryption" => {
                    builder = builder
                        .set_enable_inter_container_traffic_encryption(::aws_smithy_json::deserialize::token::expect_bool_or_null(tokens.next())?);
                }
                "EnableManagedSpotTraining" => {
                    builder = builder.set_enable_managed_spot_training(::aws_smithy_json::deserialize::token::expect_bool_or_null(tokens.next())?);
                }
                "CheckpointConfig" => {
                    builder = builder.set_checkpoint_config(crate::protocol_serde::shape_checkpoint_config::de_checkpoint_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "TrainingTimeInSeconds" => {
                    builder = builder.set_training_time_in_seconds(
                        ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
                            .map(i32::try_from)
                            .transpose()?,
                    );
                }
                "BillableTimeInSeconds" => {
                    builder = builder.set_billable_time_in_seconds(
                        ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
                            .map(i32::try_from)
                            .transpose()?,
                    );
                }
                "BillableTokenCount" => {
                    builder = builder.set_billable_token_count(
                        ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
                            .map(i64::try_from)
                            .transpose()?,
                    );
                }
                "DebugHookConfig" => {
                    builder = builder.set_debug_hook_config(crate::protocol_serde::shape_debug_hook_config::de_debug_hook_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "ExperimentConfig" => {
                    builder = builder.set_experiment_config(crate::protocol_serde::shape_experiment_config::de_experiment_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "DebugRuleConfigurations" => {
                    builder = builder.set_debug_rule_configurations(
                        crate::protocol_serde::shape_debug_rule_configurations::de_debug_rule_configurations(tokens, _value, depth + 1)?,
                    );
                }
                "TensorBoardOutputConfig" => {
                    builder = builder.set_tensor_board_output_config(
                        crate::protocol_serde::shape_tensor_board_output_config::de_tensor_board_output_config(tokens, _value, depth + 1)?,
                    );
                }
                "DebugRuleEvaluationStatuses" => {
                    builder = builder.set_debug_rule_evaluation_statuses(
                        crate::protocol_serde::shape_debug_rule_evaluation_statuses::de_debug_rule_evaluation_statuses(tokens, _value, depth + 1)?,
                    );
                }
                "ProfilerConfig" => {
                    builder = builder.set_profiler_config(crate::protocol_serde::shape_profiler_config::de_profiler_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "ProfilerRuleConfigurations" => {
                    builder = builder.set_profiler_rule_configurations(
                        crate::protocol_serde::shape_profiler_rule_configurations::de_profiler_rule_configurations(tokens, _value, depth + 1)?,
                    );
                }
                "ProfilerRuleEvaluationStatuses" => {
                    builder = builder.set_profiler_rule_evaluation_statuses(
                        crate::protocol_serde::shape_profiler_rule_evaluation_statuses::de_profiler_rule_evaluation_statuses(
                            tokens,
                            _value,
                            depth + 1,
                        )?,
                    );
                }
                "ProfilingStatus" => {
                    builder = builder.set_profiling_status(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| crate::types::ProfilingStatus::from(u.as_ref())))
                            .transpose()?,
                    );
                }
                "Environment" => {
                    builder = builder.set_environment(crate::protocol_serde::shape_training_environment_map::de_training_environment_map(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "RetryStrategy" => {
                    builder = builder.set_retry_strategy(crate::protocol_serde::shape_retry_strategy::de_retry_strategy(tokens, _value, depth + 1)?);
                }
                "RemoteDebugConfig" => {
                    builder = builder.set_remote_debug_config(crate::protocol_serde::shape_remote_debug_config::de_remote_debug_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "InfraCheckConfig" => {
                    builder = builder.set_infra_check_config(crate::protocol_serde::shape_infra_check_config::de_infra_check_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "ServerlessJobConfig" => {
                    builder = builder.set_serverless_job_config(crate::protocol_serde::shape_serverless_job_config::de_serverless_job_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "MlflowConfig" => {
                    builder = builder.set_mlflow_config(crate::protocol_serde::shape_mlflow_config::de_mlflow_config(tokens, _value, depth + 1)?);
                }
                "ModelPackageConfig" => {
                    builder = builder.set_model_package_config(crate::protocol_serde::shape_model_package_config::de_model_package_config(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "MlflowDetails" => {
                    builder = builder.set_mlflow_details(crate::protocol_serde::shape_mlflow_details::de_mlflow_details(tokens, _value, depth + 1)?);
                }
                "ProgressInfo" => {
                    builder = builder.set_progress_info(crate::protocol_serde::shape_training_progress_info::de_training_progress_info(
                        tokens,
                        _value,
                        depth + 1,
                    )?);
                }
                "OutputModelPackageArn" => {
                    builder = builder.set_output_model_package_arn(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
            },
            other => {
                return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
                    "expected object key or end object, found: {other:?}"
                )))
            }
        }
    }
    if tokens.next().is_some() {
        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
            "found more JSON tokens after completing parsing",
        ));
    }
    Ok(builder)
}