aws_sdk_sagemaker/protocol_serde/
shape_algorithm_specification.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2pub fn ser_algorithm_specification(
3    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
4    input: &crate::types::AlgorithmSpecification,
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.algorithm_name {
10        object.key("AlgorithmName").string(var_2.as_str());
11    }
12    if let Some(var_3) = &input.training_input_mode {
13        object.key("TrainingInputMode").string(var_3.as_str());
14    }
15    if let Some(var_4) = &input.metric_definitions {
16        let mut array_5 = object.key("MetricDefinitions").start_array();
17        for item_6 in var_4 {
18            {
19                #[allow(unused_mut)]
20                let mut object_7 = array_5.value().start_object();
21                crate::protocol_serde::shape_metric_definition::ser_metric_definition(&mut object_7, item_6)?;
22                object_7.finish();
23            }
24        }
25        array_5.finish();
26    }
27    if let Some(var_8) = &input.enable_sage_maker_metrics_time_series {
28        object.key("EnableSageMakerMetricsTimeSeries").boolean(*var_8);
29    }
30    if let Some(var_9) = &input.container_entrypoint {
31        let mut array_10 = object.key("ContainerEntrypoint").start_array();
32        for item_11 in var_9 {
33            {
34                array_10.value().string(item_11.as_str());
35            }
36        }
37        array_10.finish();
38    }
39    if let Some(var_12) = &input.container_arguments {
40        let mut array_13 = object.key("ContainerArguments").start_array();
41        for item_14 in var_12 {
42            {
43                array_13.value().string(item_14.as_str());
44            }
45        }
46        array_13.finish();
47    }
48    if let Some(var_15) = &input.training_image_config {
49        #[allow(unused_mut)]
50        let mut object_16 = object.key("TrainingImageConfig").start_object();
51        crate::protocol_serde::shape_training_image_config::ser_training_image_config(&mut object_16, var_15)?;
52        object_16.finish();
53    }
54    Ok(())
55}
56
57pub(crate) fn de_algorithm_specification<'a, I>(
58    tokens: &mut ::std::iter::Peekable<I>,
59) -> ::std::result::Result<Option<crate::types::AlgorithmSpecification>, ::aws_smithy_json::deserialize::error::DeserializeError>
60where
61    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
62{
63    match tokens.next().transpose()? {
64        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
65        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
66            #[allow(unused_mut)]
67            let mut builder = crate::types::builders::AlgorithmSpecificationBuilder::default();
68            loop {
69                match tokens.next().transpose()? {
70                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
71                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
72                        "TrainingImage" => {
73                            builder = builder.set_training_image(
74                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
75                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
76                                    .transpose()?,
77                            );
78                        }
79                        "AlgorithmName" => {
80                            builder = builder.set_algorithm_name(
81                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
82                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
83                                    .transpose()?,
84                            );
85                        }
86                        "TrainingInputMode" => {
87                            builder = builder.set_training_input_mode(
88                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
89                                    .map(|s| s.to_unescaped().map(|u| crate::types::TrainingInputMode::from(u.as_ref())))
90                                    .transpose()?,
91                            );
92                        }
93                        "MetricDefinitions" => {
94                            builder = builder
95                                .set_metric_definitions(crate::protocol_serde::shape_metric_definition_list::de_metric_definition_list(tokens)?);
96                        }
97                        "EnableSageMakerMetricsTimeSeries" => {
98                            builder = builder.set_enable_sage_maker_metrics_time_series(::aws_smithy_json::deserialize::token::expect_bool_or_null(
99                                tokens.next(),
100                            )?);
101                        }
102                        "ContainerEntrypoint" => {
103                            builder = builder.set_container_entrypoint(
104                                crate::protocol_serde::shape_training_container_entrypoint::de_training_container_entrypoint(tokens)?,
105                            );
106                        }
107                        "ContainerArguments" => {
108                            builder = builder.set_container_arguments(
109                                crate::protocol_serde::shape_training_container_arguments::de_training_container_arguments(tokens)?,
110                            );
111                        }
112                        "TrainingImageConfig" => {
113                            builder = builder
114                                .set_training_image_config(crate::protocol_serde::shape_training_image_config::de_training_image_config(tokens)?);
115                        }
116                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
117                    },
118                    other => {
119                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
120                            "expected object key or end object, found: {:?}",
121                            other
122                        )))
123                    }
124                }
125            }
126            Ok(Some(crate::serde_util::algorithm_specification_correct_errors(builder).build()))
127        }
128        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
129            "expected start object or null",
130        )),
131    }
132}