Handle bleed-over issue

compliance_checks/base.py

@@ -14,7 +14,11 @@ def walk_to_next_heading(card, heading, heading_text) -> bool:
         content = []
 
         sibling_gen = heading_node.nextSiblingGenerator()
-        sibling = next(sibling_gen)
+
+        try:
+            sibling = next(sibling_gen)
+        except StopIteration:
+            return False
 
         while sibling and (not (sibling.name is not None and sibling.name in stop_at) or sibling.name is None):
             if sibling.name == "p":

tests/test_intended_purpose_check.py

@@ -98,6 +98,10 @@ bloom = """\
 This model is being created in order to enable public research on large language models (LLMs).
 """
 
+bleed_over = """\
+## Uses
+"""
+
 success_result = IntendedPurposeResult(
     status=True
 )
@@ -126,3 +130,10 @@ def test_fail_on_empty_template():
     card_soup = BeautifulSoup(model_card_html, features="html.parser")
     results = IntendedPurposeCheck().run_check(card_soup)
     assert results == IntendedPurposeResult()
+
+
+def test_fail_on_bleed_over():
+    model_card_html = markdown.markdown(bleed_over)
+    card_soup = BeautifulSoup(model_card_html, features="html.parser")
+    results = IntendedPurposeCheck().run_check(card_soup)
+    assert results == IntendedPurposeResult()