aws-sdk-sagemaker 1.188.0

AWS SDK for Amazon SageMaker Service
Documentation
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
pub fn ser_hyper_parameter_training_job_definition(
    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
    input: &crate::types::HyperParameterTrainingJobDefinition,
) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
    if let Some(var_1) = &input.definition_name {
        object.key("DefinitionName").string(var_1.as_str());
    }
    if let Some(var_2) = &input.tuning_objective {
        #[allow(unused_mut)]
        let mut object_3 = object.key("TuningObjective").start_object();
        crate::protocol_serde::shape_hyper_parameter_tuning_job_objective::ser_hyper_parameter_tuning_job_objective(&mut object_3, var_2)?;
        object_3.finish();
    }
    if let Some(var_4) = &input.hyper_parameter_ranges {
        #[allow(unused_mut)]
        let mut object_5 = object.key("HyperParameterRanges").start_object();
        crate::protocol_serde::shape_parameter_ranges::ser_parameter_ranges(&mut object_5, var_4)?;
        object_5.finish();
    }
    if let Some(var_6) = &input.static_hyper_parameters {
        #[allow(unused_mut)]
        let mut object_7 = object.key("StaticHyperParameters").start_object();
        for (key_8, value_9) in var_6 {
            {
                object_7.key(key_8.as_str()).string(value_9.as_str());
            }
        }
        object_7.finish();
    }
    if let Some(var_10) = &input.algorithm_specification {
        #[allow(unused_mut)]
        let mut object_11 = object.key("AlgorithmSpecification").start_object();
        crate::protocol_serde::shape_hyper_parameter_algorithm_specification::ser_hyper_parameter_algorithm_specification(&mut object_11, var_10)?;
        object_11.finish();
    }
    if let Some(var_12) = &input.role_arn {
        object.key("RoleArn").string(var_12.as_str());
    }
    if let Some(var_13) = &input.input_data_config {
        let mut array_14 = object.key("InputDataConfig").start_array();
        for item_15 in var_13 {
            {
                #[allow(unused_mut)]
                let mut object_16 = array_14.value().start_object();
                crate::protocol_serde::shape_channel::ser_channel(&mut object_16, item_15)?;
                object_16.finish();
            }
        }
        array_14.finish();
    }
    if let Some(var_17) = &input.vpc_config {
        #[allow(unused_mut)]
        let mut object_18 = object.key("VpcConfig").start_object();
        crate::protocol_serde::shape_vpc_config::ser_vpc_config(&mut object_18, var_17)?;
        object_18.finish();
    }
    if let Some(var_19) = &input.output_data_config {
        #[allow(unused_mut)]
        let mut object_20 = object.key("OutputDataConfig").start_object();
        crate::protocol_serde::shape_output_data_config::ser_output_data_config(&mut object_20, var_19)?;
        object_20.finish();
    }
    if let Some(var_21) = &input.resource_config {
        #[allow(unused_mut)]
        let mut object_22 = object.key("ResourceConfig").start_object();
        crate::protocol_serde::shape_resource_config::ser_resource_config(&mut object_22, var_21)?;
        object_22.finish();
    }
    if let Some(var_23) = &input.hyper_parameter_tuning_resource_config {
        #[allow(unused_mut)]
        let mut object_24 = object.key("HyperParameterTuningResourceConfig").start_object();
        crate::protocol_serde::shape_hyper_parameter_tuning_resource_config::ser_hyper_parameter_tuning_resource_config(&mut object_24, var_23)?;
        object_24.finish();
    }
    if let Some(var_25) = &input.stopping_condition {
        #[allow(unused_mut)]
        let mut object_26 = object.key("StoppingCondition").start_object();
        crate::protocol_serde::shape_stopping_condition::ser_stopping_condition(&mut object_26, var_25)?;
        object_26.finish();
    }
    if let Some(var_27) = &input.enable_network_isolation {
        object.key("EnableNetworkIsolation").boolean(*var_27);
    }
    if let Some(var_28) = &input.enable_inter_container_traffic_encryption {
        object.key("EnableInterContainerTrafficEncryption").boolean(*var_28);
    }
    if let Some(var_29) = &input.enable_managed_spot_training {
        object.key("EnableManagedSpotTraining").boolean(*var_29);
    }
    if let Some(var_30) = &input.checkpoint_config {
        #[allow(unused_mut)]
        let mut object_31 = object.key("CheckpointConfig").start_object();
        crate::protocol_serde::shape_checkpoint_config::ser_checkpoint_config(&mut object_31, var_30)?;
        object_31.finish();
    }
    if let Some(var_32) = &input.retry_strategy {
        #[allow(unused_mut)]
        let mut object_33 = object.key("RetryStrategy").start_object();
        crate::protocol_serde::shape_retry_strategy::ser_retry_strategy(&mut object_33, var_32)?;
        object_33.finish();
    }
    if let Some(var_34) = &input.environment {
        #[allow(unused_mut)]
        let mut object_35 = object.key("Environment").start_object();
        for (key_36, value_37) in var_34 {
            {
                object_35.key(key_36.as_str()).string(value_37.as_str());
            }
        }
        object_35.finish();
    }
    Ok(())
}

pub(crate) fn de_hyper_parameter_training_job_definition<'a, I>(
    tokens: &mut ::std::iter::Peekable<I>,
    _value: &'a [u8],
) -> ::std::result::Result<Option<crate::types::HyperParameterTrainingJobDefinition>, ::aws_smithy_json::deserialize::error::DeserializeError>
where
    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
{
    match tokens.next().transpose()? {
        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
            #[allow(unused_mut)]
            let mut builder = crate::types::builders::HyperParameterTrainingJobDefinitionBuilder::default();
            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() {
                        "DefinitionName" => {
                            builder = builder.set_definition_name(
                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                                    .transpose()?,
                            );
                        }
                        "TuningObjective" => {
                            builder = builder.set_tuning_objective(
                                crate::protocol_serde::shape_hyper_parameter_tuning_job_objective::de_hyper_parameter_tuning_job_objective(
                                    tokens, _value,
                                )?,
                            );
                        }
                        "HyperParameterRanges" => {
                            builder = builder
                                .set_hyper_parameter_ranges(crate::protocol_serde::shape_parameter_ranges::de_parameter_ranges(tokens, _value)?);
                        }
                        "StaticHyperParameters" => {
                            builder = builder
                                .set_static_hyper_parameters(crate::protocol_serde::shape_hyper_parameters::de_hyper_parameters(tokens, _value)?);
                        }
                        "AlgorithmSpecification" => {
                            builder = builder.set_algorithm_specification(
                                crate::protocol_serde::shape_hyper_parameter_algorithm_specification::de_hyper_parameter_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)?);
                        }
                        "VpcConfig" => {
                            builder = builder.set_vpc_config(crate::protocol_serde::shape_vpc_config::de_vpc_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)?);
                        }
                        "HyperParameterTuningResourceConfig" => {
                            builder = builder.set_hyper_parameter_tuning_resource_config(
                                crate::protocol_serde::shape_hyper_parameter_tuning_resource_config::de_hyper_parameter_tuning_resource_config(
                                    tokens, _value,
                                )?,
                            );
                        }
                        "StoppingCondition" => {
                            builder = builder
                                .set_stopping_condition(crate::protocol_serde::shape_stopping_condition::de_stopping_condition(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)?);
                        }
                        "RetryStrategy" => {
                            builder = builder.set_retry_strategy(crate::protocol_serde::shape_retry_strategy::de_retry_strategy(tokens, _value)?);
                        }
                        "Environment" => {
                            builder = builder.set_environment(
                                    crate::protocol_serde::shape_hyper_parameter_training_job_environment_map::de_hyper_parameter_training_job_environment_map(tokens, _value)?
                                );
                        }
                        _ => ::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:?}"
                        )))
                    }
                }
            }
            Ok(Some(
                crate::serde_util::hyper_parameter_training_job_definition_correct_errors(builder).build(),
            ))
        }
        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
            "expected start object or null",
        )),
    }
}