aws_sdk_sagemaker/protocol_serde/
shape_training_specification.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2pub fn ser_training_specification(
3    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
4    input: &crate::types::TrainingSpecification,
5) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
6    if let Some(var_1) = &input.training_image {
7        object.key("TrainingImage").string(var_1.as_str());
8    }
9    if let Some(var_2) = &input.training_image_digest {
10        object.key("TrainingImageDigest").string(var_2.as_str());
11    }
12    if let Some(var_3) = &input.supported_hyper_parameters {
13        let mut array_4 = object.key("SupportedHyperParameters").start_array();
14        for item_5 in var_3 {
15            {
16                #[allow(unused_mut)]
17                let mut object_6 = array_4.value().start_object();
18                crate::protocol_serde::shape_hyper_parameter_specification::ser_hyper_parameter_specification(&mut object_6, item_5)?;
19                object_6.finish();
20            }
21        }
22        array_4.finish();
23    }
24    if let Some(var_7) = &input.supported_training_instance_types {
25        let mut array_8 = object.key("SupportedTrainingInstanceTypes").start_array();
26        for item_9 in var_7 {
27            {
28                array_8.value().string(item_9.as_str());
29            }
30        }
31        array_8.finish();
32    }
33    if let Some(var_10) = &input.supports_distributed_training {
34        object.key("SupportsDistributedTraining").boolean(*var_10);
35    }
36    if let Some(var_11) = &input.metric_definitions {
37        let mut array_12 = object.key("MetricDefinitions").start_array();
38        for item_13 in var_11 {
39            {
40                #[allow(unused_mut)]
41                let mut object_14 = array_12.value().start_object();
42                crate::protocol_serde::shape_metric_definition::ser_metric_definition(&mut object_14, item_13)?;
43                object_14.finish();
44            }
45        }
46        array_12.finish();
47    }
48    if let Some(var_15) = &input.training_channels {
49        let mut array_16 = object.key("TrainingChannels").start_array();
50        for item_17 in var_15 {
51            {
52                #[allow(unused_mut)]
53                let mut object_18 = array_16.value().start_object();
54                crate::protocol_serde::shape_channel_specification::ser_channel_specification(&mut object_18, item_17)?;
55                object_18.finish();
56            }
57        }
58        array_16.finish();
59    }
60    if let Some(var_19) = &input.supported_tuning_job_objective_metrics {
61        let mut array_20 = object.key("SupportedTuningJobObjectiveMetrics").start_array();
62        for item_21 in var_19 {
63            {
64                #[allow(unused_mut)]
65                let mut object_22 = array_20.value().start_object();
66                crate::protocol_serde::shape_hyper_parameter_tuning_job_objective::ser_hyper_parameter_tuning_job_objective(&mut object_22, item_21)?;
67                object_22.finish();
68            }
69        }
70        array_20.finish();
71    }
72    if let Some(var_23) = &input.additional_s3_data_source {
73        #[allow(unused_mut)]
74        let mut object_24 = object.key("AdditionalS3DataSource").start_object();
75        crate::protocol_serde::shape_additional_s3_data_source::ser_additional_s3_data_source(&mut object_24, var_23)?;
76        object_24.finish();
77    }
78    Ok(())
79}
80
81pub(crate) fn de_training_specification<'a, I>(
82    tokens: &mut ::std::iter::Peekable<I>,
83) -> ::std::result::Result<Option<crate::types::TrainingSpecification>, ::aws_smithy_json::deserialize::error::DeserializeError>
84where
85    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
86{
87    match tokens.next().transpose()? {
88        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
89        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
90            #[allow(unused_mut)]
91            let mut builder = crate::types::builders::TrainingSpecificationBuilder::default();
92            loop {
93                match tokens.next().transpose()? {
94                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
95                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
96                        "TrainingImage" => {
97                            builder = builder.set_training_image(
98                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
99                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
100                                    .transpose()?,
101                            );
102                        }
103                        "TrainingImageDigest" => {
104                            builder = builder.set_training_image_digest(
105                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
106                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
107                                    .transpose()?,
108                            );
109                        }
110                        "SupportedHyperParameters" => {
111                            builder = builder.set_supported_hyper_parameters(
112                                crate::protocol_serde::shape_hyper_parameter_specifications::de_hyper_parameter_specifications(tokens)?,
113                            );
114                        }
115                        "SupportedTrainingInstanceTypes" => {
116                            builder = builder.set_supported_training_instance_types(
117                                crate::protocol_serde::shape_training_instance_types::de_training_instance_types(tokens)?,
118                            );
119                        }
120                        "SupportsDistributedTraining" => {
121                            builder =
122                                builder.set_supports_distributed_training(::aws_smithy_json::deserialize::token::expect_bool_or_null(tokens.next())?);
123                        }
124                        "MetricDefinitions" => {
125                            builder = builder
126                                .set_metric_definitions(crate::protocol_serde::shape_metric_definition_list::de_metric_definition_list(tokens)?);
127                        }
128                        "TrainingChannels" => {
129                            builder = builder
130                                .set_training_channels(crate::protocol_serde::shape_channel_specifications::de_channel_specifications(tokens)?);
131                        }
132                        "SupportedTuningJobObjectiveMetrics" => {
133                            builder = builder.set_supported_tuning_job_objective_metrics(
134                                crate::protocol_serde::shape_hyper_parameter_tuning_job_objectives::de_hyper_parameter_tuning_job_objectives(tokens)?,
135                            );
136                        }
137                        "AdditionalS3DataSource" => {
138                            builder = builder.set_additional_s3_data_source(
139                                crate::protocol_serde::shape_additional_s3_data_source::de_additional_s3_data_source(tokens)?,
140                            );
141                        }
142                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
143                    },
144                    other => {
145                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
146                            "expected object key or end object, found: {:?}",
147                            other
148                        )))
149                    }
150                }
151            }
152            Ok(Some(crate::serde_util::training_specification_correct_errors(builder).build()))
153        }
154        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
155            "expected start object or null",
156        )),
157    }
158}
aws_sdk_sagemaker/protocol_serde/shape_training_specification.rs

aws_sdk_sagemaker/protocol_serde/
shape_training_specification.rs
