apply fix to type-annotated lists

crates/ruff_linter/src/rules/perflint/rules/manual_list_comprehension.rs

@@ -252,7 +252,6 @@ pub(crate) fn manual_list_comprehension(checker: &mut Checker, for_stmt: &ast::S
         ast::Stmt::Assign(assign) => Some(&assign.value),
         _ => None,
     });
-    dbg!(binding_value);
     // If the variable is an empty list literal, then we might be able to replace it with a full list comprehension
     // otherwise, it has to be replaced with a `list.extend`
     let binding_is_empty_list =
@@ -413,8 +412,16 @@ fn convert_to_list_extend(
                 .ok_or(anyhow!(
                     "Binding must have a statement to convert into a list comprehension"
                 ))?;
-            let mut comments_to_move =
-                comment_strings_in_range(binding_stmt_range);
+
+            let annotations = match binding
+                .statement(checker.semantic())
+                .and_then(|stmt| stmt.as_ann_assign_stmt())
+            {
+                Some(assign) => format!(": {}", locator.slice(assign.annotation.range())),
+                None => String::new(),
+            };
+
+            let mut comments_to_move = comment_strings_in_range(binding_stmt_range);
             comments_to_move.extend(for_loop_inline_comments);
 
             let indentation = if comments_to_move.is_empty() {
@@ -425,7 +432,7 @@ fn convert_to_list_extend(
             let leading_comments = format!("{}{indentation}", comments_to_move.join(&indentation));
 
             let comprehension_body =
-                format!("{leading_comments}{variable_name} = [{generator_str}]");
+                format!("{leading_comments}{variable_name}{annotations} = [{generator_str}]");
 
             Ok(Fix::unsafe_edits(
                 Edit::range_deletion(binding_stmt_range),

crates/ruff_linter/src/rules/perflint/snapshots/ruff_linter__rules__perflint__tests__PERF401_PERF401.py.snap

@@ -149,5 +149,16 @@ PERF401.py:175:9: PERF401 Use a list comprehension to create a transformed list
 180 | |             )
 181 | |         ) # PERF401
     | |_________^ PERF401
+182 |   
+183 |   def f():
+    |
+    = help: Replace for loop with list comprehension
+
+PERF401.py:186:9: PERF401 Use a list comprehension to create a transformed list
+    |
+184 |     result: list[int] = []
+185 |     for i in range(10):
+186 |         result.append(i*2) # PERF401
+    |         ^^^^^^^^^^^^^^^^^^ PERF401
     |
     = help: Replace for loop with list comprehension

crates/ruff_linter/src/rules/perflint/snapshots/ruff_linter__rules__perflint__tests__preview__PERF401_PERF401.py.snap

@@ -330,6 +330,8 @@ PERF401.py:175:9: PERF401 [*] Use a list comprehension to create a transformed l
 180 | |             )
 181 | |         ) # PERF401
     | |_________^ PERF401
+182 |   
+183 |   def f():
     |
     = help: Replace for loop with list comprehension
 
@@ -348,3 +350,24 @@ PERF401.py:175:9: PERF401 [*] Use a list comprehension to create a transformed l
 180     |-            )
 181     |-        ) # PERF401
     177 |+            ) for i in range(2)] # PERF401
+182 178 | 
+183 179 | def f():
+184 180 |     result: list[int] = []
+
+PERF401.py:186:9: PERF401 [*] Use a list comprehension to create a transformed list
+    |
+184 |     result: list[int] = []
+185 |     for i in range(10):
+186 |         result.append(i*2) # PERF401
+    |         ^^^^^^^^^^^^^^^^^^ PERF401
+    |
+    = help: Replace for loop with list comprehension
+
+ℹ Unsafe fix
+181 181 |         ) # PERF401
+182 182 | 
+183 183 | def f():
+184     |-    result: list[int] = []
+185     |-    for i in range(10):
+186     |-        result.append(i*2) # PERF401
+    184 |+    result: list[int] = [i*2 for i in range(10)] # PERF401