special raise

astral-sh · Apr 17, 2024 · 8704a3f · 8704a3f
1 parent 451a2f9
commit 8704a3f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/crates/ruff_linter/src/rules/pylint/rules/invalid_index_return.rs b/crates/ruff_linter/src/rules/pylint/rules/invalid_index_return.rs
@@ -63,6 +63,17 @@ pub(crate) fn invalid_index_return(checker: &mut Checker, name: &str, body: &[St
         return;
     }
 
+    let body_without_comments = body
+        .iter()
+        .filter(|stmt| !matches!(stmt, Stmt::Expr(expr) if expr.value.is_string_literal_expr()))
+        .collect::<Vec<_>>();
+    if body_without_comments.is_empty() {
+        return;
+    }
+    if body_without_comments.len() == 1 && body_without_comments[0].is_raise_stmt() {
+        return;
+    }
+
     let returns = {
         let mut visitor = ReturnStatementVisitor::default();
         visitor.visit_body(body);