Slightly nicer code

maxall41 · Feb 12, 2024 · 3eda7ff · 3eda7ff
1 parent 7af62fe
commit 3eda7ff
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -154,8 +154,7 @@ fn main() {
             }
         }
     }
-    if args.pdb_output_path.is_some() {
-        let pdb_path = args.pdb_output_path.unwrap();
+    if let Some(pdb_path) = args.pdb_output_path {
         pdbtbx::save(&pdb, &pdb_path, pdbtbx::StrictnessLevel::Loose).unwrap();
         println!(
             "{}",
@@ -166,8 +165,7 @@ fn main() {
             .green()
         );
     }
-    if args.csv_output_path.is_some() {
-        let csv_path = args.csv_output_path.unwrap();
+    if let Some(csv_path) = args.csv_output_path {
         let mut df: DataFrame = df!(
             "Residue ID" => &dpx_residue_ids,
             "DPX" => &dpx_values