Format module level docstring

astral-sh · Jan 30, 2024 · fc22e42 · fc22e42
1 parent 7329bf4
commit fc22e42
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
@@ -138,7 +138,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);