Use single filter call

Author: MichaReiser

parser/src/parser.rs

@@ -415,13 +415,22 @@ pub fn parse_tokens(
 ) -> Result<ast::Mod, ParseError> {
     let lxr = lxr.into_iter();
 
-    let lxr =
-        lxr.filter_ok(|(tok, _)| !matches!(tok, Tok::Comment { .. } | Tok::NonLogicalNewline));
-    if mode == Mode::Jupyter {
-        let lxr = lxr.filter_ok(|(tok, _)| !matches!(tok, Tok::MagicCommand { .. }));
-        parse_filtered_tokens(lxr, mode, source_path)
-    } else {
-        parse_filtered_tokens(lxr, mode, source_path)
+    match mode {
+        Mode::Module | Mode::Interactive | Mode::Expression => parse_filtered_tokens(
+            lxr.filter_ok(|(tok, _)| !matches!(tok, Tok::Comment { .. } | Tok::NonLogicalNewline)),
+            mode,
+            source_path,
+        ),
+        Mode::Jupyter => parse_filtered_tokens(
+            lxr.filter_ok(|(tok, _)| {
+                !matches!(
+                    tok,
+                    Tok::Comment { .. } | Tok::NonLogicalNewline | Tok::MagicCommand { .. }
+                )
+            }),
+            mode,
+            source_path,
+        ),
     }
 }