Format module level docstring

astral-sh · Jan 30, 2024 · bef24c2 · bef24c2
1 parent 6bb1264
commit bef24c2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/crates/ruff_python_formatter/src/statement/suite.rs b/crates/ruff_python_formatter/src/statement/suite.rs
@@ -140,7 +140,13 @@ impl FormatRule<Suite, PyFormatContext<'_>> for FormatSuite {
                     SuiteChildStatement::Other(first)
                 }
             }
-            SuiteKind::TopLevel => SuiteChildStatement::Other(first),
+            SuiteKind::TopLevel => {
+                if let Some(docstring) = DocstringStmt::try_from_statement(first, self.kind) {
+                    SuiteChildStatement::Docstring(docstring)
+                } else {
+                    SuiteChildStatement::Other(first)
+                }
+            }
         };
 
         let first_comments = comments.leading_dangling_trailing(first);