tidy: remove duplicate comments

Author: dmadisetti

marimo/_ast/transformers.py

@@ -195,8 +195,11 @@ def __init__(self, import_name: str, keep_one: bool = False) -> None:
         self.keep_one = keep_one
         self.import_name = import_name
 
-    def _return_once(self, node: ast.Import | ast.ImportFrom, original_names:
-                     list[ast.alias]) -> Optional[ast.Import | ast.ImportFrom]:
+    def _return_once(
+        self,
+        node: ast.Import | ast.ImportFrom,
+        original_names: list[ast.alias],
+    ) -> Optional[ast.Import | ast.ImportFrom]:
         if node.names:
             return node
         elif self.keep_one:

marimo/_convert/comment_preserver.py

@@ -73,7 +73,9 @@ def wrapper(*args: object, **kwargs: object) -> list[str]:
             original_sources = args[0]
 
             # Merge comments back into transformed sources
-            result = self._merge_comments(original_sources, transformed_sources)
+            result = self._merge_comments(
+                original_sources, transformed_sources
+            )
 
             # Update our internal comment data to track only the clean transformed sources
             # This clears old comments that no longer apply
@@ -158,10 +160,14 @@ def _apply_comments_to_source(
 
             if chosen_comment:
                 comment_text = chosen_comment.text
-                if chosen_comment.col > 0 and target_line_idx < len(original_lines):
+                if chosen_comment.col > 0 and target_line_idx < len(
+                    original_lines
+                ):
                     # Inline comment - append to the line if not already present
                     current_line = result_lines[target_line_idx]
-                    if not current_line.rstrip().endswith(comment_text.rstrip()):
+                    if not current_line.rstrip().endswith(
+                        comment_text.rstrip()
+                    ):
                         result_lines[target_line_idx] = (
                             current_line.rstrip() + "  " + comment_text
                         )
@@ -171,7 +177,9 @@ def _apply_comments_to_source(
 
         return "\n".join(result_lines)
 
-    def _update_comments_for_transformed_sources(self, sources: list[str]) -> None:
+    def _update_comments_for_transformed_sources(
+        self, sources: list[str]
+    ) -> None:
         """Update internal comment data to track the transformed sources."""
         self.sources = sources
         self.comments_by_source = {}

marimo/_convert/ipynb.py

@@ -659,10 +659,14 @@ def transform_remove_duplicate_imports(sources: list[str]) -> list[str]:
         scoped = set()
         for var, instances in cell.variable_data.items():
             for instance in instances:
-                if (var in imports or var in scoped) and instance.kind == "import":
+                if (
+                    var in imports or var in scoped
+                ) and instance.kind == "import":
                     # If it's not in global imports, we keep one instance
                     keep_one = var not in imports
-                    transformer = RemoveImportTransformer(var, keep_one=keep_one)
+                    transformer = RemoveImportTransformer(
+                        var, keep_one=keep_one
+                    )
                     source = transformer.strip_imports(source)
                 scoped.add(var)
         imports.update(scoped)