Update docs_formatter.py

ivy-llc · Nov 29, 2023 · 33cd752 · 33cd752
1 parent a68e93b
commit 33cd752
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ivy_lint/formatters/docs_formatter.py b/ivy_lint/formatters/docs_formatter.py
@@ -31,6 +31,8 @@ def format_docstring(self, doc):
             elif is_codeblock and not is_codeblock_cont and (not stripped_line or (not stripped_line.startswith(('>>>', '...')))):
                 is_codeblock = False
             if is_codeblock:
+                if stripped_line.startswith(('>>>')):
+                    lb = rb = 0
                 lb += line.count('(')
                 rb += line.count(')')
                 if rb >= lb: