Ignore F401 in __init__.py

Zac-HD · Apr 25, 2024 · 301523b · 301523b
1 parent f56ab96
commit 301523b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/shed/_cli.py b/src/shed/_cli.py
@@ -84,6 +84,8 @@ def _rewrite_on_disk(
         writer: Callable[..., str] = docshed
     elif fname.endswith(".pyi"):
         writer = functools.partial(shed, is_pyi=True)
+    elif Path(fname).name == "__init__.py":
+        writer = functools.partial(shed, _remove_unused_imports=False)
     else:
         writer = shed