Skip to content

Commit b5121b3

Browse files
committed
Small idiomatic change, using map_err instead of match.
1 parent b9b84b7 commit b5121b3

File tree

1 file changed

+5
-10
lines changed

1 file changed

+5
-10
lines changed

src/saved_model.rs

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -495,16 +495,11 @@ impl SavedModelSaver {
495495
) -> std::result::Result<(), SaveModelError> {
496496
let mut meta_graph = self.meta_graph.clone();
497497
let graph_bytes = graph.graph_def()?;
498-
let graph_def = match protobuf::Message::parse_from_bytes(&graph_bytes) {
499-
Ok(x) => x,
500-
Err(e) => {
501-
return Err(Status::new_set_lossy(
502-
Code::InvalidArgument,
503-
&format!("Unable to parse graph definition: {}", e),
504-
)
505-
.into());
506-
}
507-
};
498+
let graph_def= protobuf::Message::parse_from_bytes(&graph_bytes)
499+
.map_err(|e| <Status as Into<SaveModelError>>::into(Status::new_set_lossy(
500+
Code::InvalidArgument,
501+
&format!("Unable to parse graph definition: {}", e),
502+
)))?;
508503
meta_graph.set_graph_def(graph_def);
509504
let mut saved_model = protos::saved_model::SavedModel::new();
510505
saved_model.set_saved_model_schema_version(1);

0 commit comments

Comments
 (0)