refactor: use retain for error messages

lambdaclass · Jul 31, 2023 · f682898 · f682898
1 parent cdf9880
commit f682898
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/services/api/contract_classes/deprecated_contract_class.rs b/src/services/api/contract_classes/deprecated_contract_class.rs
@@ -238,11 +238,9 @@ pub(crate) fn to_cairo_runner_program(
         return Err(ProgramError::PrimeDiffers(program.prime.to_string()));
     };
 
-    let error_message_attributes = serde_json::from_value::<Vec<Attribute>>(program.attributes)
-        .unwrap_or(Vec::new())
-        .into_iter()
-        .filter(|attr| attr.name == "error_message")
-        .collect();
+    let mut error_message_attributes =
+        serde_json::from_value::<Vec<Attribute>>(program.attributes).unwrap_or_default();
+    error_message_attributes.retain(|attr| attr.name == "error_message");
 
     let program = Program::new(
         serde_json::from_value::<Vec<BuiltinName>>(program.builtins)?,