aws_sdk_sagemaker/protocol_serde/
shape_hyper_parameter_tuning_job_config.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2pub fn ser_hyper_parameter_tuning_job_config(
3    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
4    input: &crate::types::HyperParameterTuningJobConfig,
5) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
6    if let Some(var_1) = &input.strategy {
7        object.key("Strategy").string(var_1.as_str());
8    }
9    if let Some(var_2) = &input.strategy_config {
10        #[allow(unused_mut)]
11        let mut object_3 = object.key("StrategyConfig").start_object();
12        crate::protocol_serde::shape_hyper_parameter_tuning_job_strategy_config::ser_hyper_parameter_tuning_job_strategy_config(
13            &mut object_3,
14            var_2,
15        )?;
16        object_3.finish();
17    }
18    if let Some(var_4) = &input.hyper_parameter_tuning_job_objective {
19        #[allow(unused_mut)]
20        let mut object_5 = object.key("HyperParameterTuningJobObjective").start_object();
21        crate::protocol_serde::shape_hyper_parameter_tuning_job_objective::ser_hyper_parameter_tuning_job_objective(&mut object_5, var_4)?;
22        object_5.finish();
23    }
24    if let Some(var_6) = &input.resource_limits {
25        #[allow(unused_mut)]
26        let mut object_7 = object.key("ResourceLimits").start_object();
27        crate::protocol_serde::shape_resource_limits::ser_resource_limits(&mut object_7, var_6)?;
28        object_7.finish();
29    }
30    if let Some(var_8) = &input.parameter_ranges {
31        #[allow(unused_mut)]
32        let mut object_9 = object.key("ParameterRanges").start_object();
33        crate::protocol_serde::shape_parameter_ranges::ser_parameter_ranges(&mut object_9, var_8)?;
34        object_9.finish();
35    }
36    if let Some(var_10) = &input.training_job_early_stopping_type {
37        object.key("TrainingJobEarlyStoppingType").string(var_10.as_str());
38    }
39    if let Some(var_11) = &input.tuning_job_completion_criteria {
40        #[allow(unused_mut)]
41        let mut object_12 = object.key("TuningJobCompletionCriteria").start_object();
42        crate::protocol_serde::shape_tuning_job_completion_criteria::ser_tuning_job_completion_criteria(&mut object_12, var_11)?;
43        object_12.finish();
44    }
45    if let Some(var_13) = &input.random_seed {
46        object.key("RandomSeed").number(
47            #[allow(clippy::useless_conversion)]
48            ::aws_smithy_types::Number::NegInt((*var_13).into()),
49        );
50    }
51    Ok(())
52}
53
54pub(crate) fn de_hyper_parameter_tuning_job_config<'a, I>(
55    tokens: &mut ::std::iter::Peekable<I>,
56) -> ::std::result::Result<Option<crate::types::HyperParameterTuningJobConfig>, ::aws_smithy_json::deserialize::error::DeserializeError>
57where
58    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
59{
60    match tokens.next().transpose()? {
61        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
62        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
63            #[allow(unused_mut)]
64            let mut builder = crate::types::builders::HyperParameterTuningJobConfigBuilder::default();
65            loop {
66                match tokens.next().transpose()? {
67                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
68                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
69                        "Strategy" => {
70                            builder = builder.set_strategy(
71                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
72                                    .map(|s| {
73                                        s.to_unescaped()
74                                            .map(|u| crate::types::HyperParameterTuningJobStrategyType::from(u.as_ref()))
75                                    })
76                                    .transpose()?,
77                            );
78                        }
79                        "StrategyConfig" => {
80                            builder = builder.set_strategy_config(
81                                    crate::protocol_serde::shape_hyper_parameter_tuning_job_strategy_config::de_hyper_parameter_tuning_job_strategy_config(tokens)?
82                                );
83                        }
84                        "HyperParameterTuningJobObjective" => {
85                            builder = builder.set_hyper_parameter_tuning_job_objective(
86                                crate::protocol_serde::shape_hyper_parameter_tuning_job_objective::de_hyper_parameter_tuning_job_objective(tokens)?,
87                            );
88                        }
89                        "ResourceLimits" => {
90                            builder = builder.set_resource_limits(crate::protocol_serde::shape_resource_limits::de_resource_limits(tokens)?);
91                        }
92                        "ParameterRanges" => {
93                            builder = builder.set_parameter_ranges(crate::protocol_serde::shape_parameter_ranges::de_parameter_ranges(tokens)?);
94                        }
95                        "TrainingJobEarlyStoppingType" => {
96                            builder = builder.set_training_job_early_stopping_type(
97                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
98                                    .map(|s| s.to_unescaped().map(|u| crate::types::TrainingJobEarlyStoppingType::from(u.as_ref())))
99                                    .transpose()?,
100                            );
101                        }
102                        "TuningJobCompletionCriteria" => {
103                            builder = builder.set_tuning_job_completion_criteria(
104                                crate::protocol_serde::shape_tuning_job_completion_criteria::de_tuning_job_completion_criteria(tokens)?,
105                            );
106                        }
107                        "RandomSeed" => {
108                            builder = builder.set_random_seed(
109                                ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
110                                    .map(i32::try_from)
111                                    .transpose()?,
112                            );
113                        }
114                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
115                    },
116                    other => {
117                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
118                            "expected object key or end object, found: {:?}",
119                            other
120                        )))
121                    }
122                }
123            }
124            Ok(Some(crate::serde_util::hyper_parameter_tuning_job_config_correct_errors(builder).build()))
125        }
126        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
127            "expected start object or null",
128        )),
129    }
130}