rust: remove unnecessary unwraps

Signed-off-by: William Woodruff <william@trailofbits.com>
trailofbits · Oct 1, 2024 · 5874af1 · 5874af1
1 parent 14e1c5b
commit 5874af1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
@@ -62,7 +62,7 @@ impl TimeStampReq {
             contents: OwnedMessageImprint::try_new(self.raw.borrow_owner().clone_ref(py), |v| {
                 RawMessageImprint::parse_data(v.as_bytes(py))
             })
-            .unwrap(),
+            .map_err(|_| PyValueError::new_err("invalid message imprint"))?,
         })
     }
 
@@ -342,7 +342,7 @@ impl PyTSTInfo {
                 self.raw.borrow_owner().clone().borrow_owner().clone_ref(py),
                 |v| RawMessageImprint::parse_data(v.as_bytes(py)),
             )
-            .unwrap(),
+            .map_err(|_| PyValueError::new_err("invalid message imprint"))?,
         })
     }