Consider the case of a Yields section

AloizioMacedo · Jun 24, 2024 · 8853d12 · 8853d12
1 parent 34e67f6
commit 8853d12
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/parsing.rs b/src/parsing.rs
@@ -44,6 +44,10 @@ pub fn parse_google_docstring(
 ) -> Option<Vec<(&str, Option<&str>)>> {
     let (_, mut args) = text.split_once("Args:\n")?;
 
+    if let Some(c) = args.find("Yields:\n") {
+        args = &args[..c];
+    };
+
     if let Some(c) = args.find("Returns:\n") {
         args = &args[..c];
     };