aws-sdk-sagemaker 1.190.0

AWS SDK for Amazon SageMaker Service
Documentation
// 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;
    ::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)?);
                }
                "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)?);
                }
                "AlgorithmSpecification" => {
                    builder = builder.set_algorithm_specification(crate::protocol_serde::shape_algorithm_specification::de_algorithm_specification(
                        tokens, _value,
                    )?);
                }
                "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)?);
                }
                "OutputDataConfig" => {
                    builder = builder.set_output_data_config(crate::protocol_serde::shape_output_data_config::de_output_data_config(tokens, _value)?);
                }
                "ResourceConfig" => {
                    builder = builder.set_resource_config(crate::protocol_serde::shape_resource_config::de_resource_config(tokens, _value)?);
                }
                "WarmPoolStatus" => {
                    builder = builder.set_warm_pool_status(crate::protocol_serde::shape_warm_pool_status::de_warm_pool_status(tokens, _value)?);
                }
                "VpcConfig" => {
                    builder = builder.set_vpc_config(crate::protocol_serde::shape_vpc_config::de_vpc_config(tokens, _value)?);
                }
                "StoppingCondition" => {
                    builder = builder.set_stopping_condition(crate::protocol_serde::shape_stopping_condition::de_stopping_condition(tokens, _value)?);
                }
                "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)?,
                    );
                }
                "FinalMetricDataList" => {
                    builder = builder.set_final_metric_data_list(crate::protocol_serde::shape_final_metric_data_list::de_final_metric_data_list(
                        tokens, _value,
                    )?);
                }
                "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)?);
                }
                "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)?);
                }
                "ExperimentConfig" => {
                    builder = builder.set_experiment_config(crate::protocol_serde::shape_experiment_config::de_experiment_config(tokens, _value)?);
                }
                "DebugRuleConfigurations" => {
                    builder = builder.set_debug_rule_configurations(
                        crate::protocol_serde::shape_debug_rule_configurations::de_debug_rule_configurations(tokens, _value)?,
                    );
                }
                "TensorBoardOutputConfig" => {
                    builder = builder.set_tensor_board_output_config(
                        crate::protocol_serde::shape_tensor_board_output_config::de_tensor_board_output_config(tokens, _value)?,
                    );
                }
                "DebugRuleEvaluationStatuses" => {
                    builder = builder.set_debug_rule_evaluation_statuses(
                        crate::protocol_serde::shape_debug_rule_evaluation_statuses::de_debug_rule_evaluation_statuses(tokens, _value)?,
                    );
                }
                "ProfilerConfig" => {
                    builder = builder.set_profiler_config(crate::protocol_serde::shape_profiler_config::de_profiler_config(tokens, _value)?);
                }
                "ProfilerRuleConfigurations" => {
                    builder = builder.set_profiler_rule_configurations(
                        crate::protocol_serde::shape_profiler_rule_configurations::de_profiler_rule_configurations(tokens, _value)?,
                    );
                }
                "ProfilerRuleEvaluationStatuses" => {
                    builder = builder.set_profiler_rule_evaluation_statuses(
                        crate::protocol_serde::shape_profiler_rule_evaluation_statuses::de_profiler_rule_evaluation_statuses(tokens, _value)?,
                    );
                }
                "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,
                    )?);
                }
                "RetryStrategy" => {
                    builder = builder.set_retry_strategy(crate::protocol_serde::shape_retry_strategy::de_retry_strategy(tokens, _value)?);
                }
                "RemoteDebugConfig" => {
                    builder =
                        builder.set_remote_debug_config(crate::protocol_serde::shape_remote_debug_config::de_remote_debug_config(tokens, _value)?);
                }
                "InfraCheckConfig" => {
                    builder = builder.set_infra_check_config(crate::protocol_serde::shape_infra_check_config::de_infra_check_config(tokens, _value)?);
                }
                "ServerlessJobConfig" => {
                    builder = builder.set_serverless_job_config(crate::protocol_serde::shape_serverless_job_config::de_serverless_job_config(
                        tokens, _value,
                    )?);
                }
                "MlflowConfig" => {
                    builder = builder.set_mlflow_config(crate::protocol_serde::shape_mlflow_config::de_mlflow_config(tokens, _value)?);
                }
                "ModelPackageConfig" => {
                    builder = builder.set_model_package_config(crate::protocol_serde::shape_model_package_config::de_model_package_config(
                        tokens, _value,
                    )?);
                }
                "MlflowDetails" => {
                    builder = builder.set_mlflow_details(crate::protocol_serde::shape_mlflow_details::de_mlflow_details(tokens, _value)?);
                }
                "ProgressInfo" => {
                    builder = builder.set_progress_info(crate::protocol_serde::shape_training_progress_info::de_training_progress_info(
                        tokens, _value,
                    )?);
                }
                "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)
}