Update docs_formatter.py

ivy-llc · Dec 1, 2023 · 8ecec4c · 8ecec4c
1 parent 306b148
commit 8ecec4c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ivy_lint/formatters/docs_formatter.py b/ivy_lint/formatters/docs_formatter.py
@@ -9,8 +9,6 @@ class DocstringFormatter(BaseDocstringFormatter):
     def validate_section_name(self, section_name, VALID_SECTION_NAMES):
         if section_name not in VALID_SECTION_NAMES:
             raise ValueError(f"Invalid section name: {section_name}. Valid section names are {VALID_SECTION_NAMES}")
-            return False
-        return True
 
     def format_docstring(self, doc):
         """Formats a single docstring."""
@@ -40,7 +38,9 @@ def format_docstring(self, doc):
 
             if stripped_line.startswith('-') and stripped_line.endswith('-'):
                 section_title = prev_line
-                if not self.validate_section_name(section_title, VALID_SECTION_NAMES):
+                try:
+                    self.validate_section_name(section_title, VALID_SECTION_NAMES)
+                except ValueError as e:
                     incorrect_sections.add(idx)
 
             if not is_codeblock and stripped_line.startswith('>>>'):