diff --git a/scripts/parse_html_for_errors.py b/scripts/parse_html_for_errors.py index df4453a..57bd719 100644 --- a/scripts/parse_html_for_errors.py +++ b/scripts/parse_html_for_errors.py @@ -28,7 +28,7 @@ def main(): # Remove div if it has an error for div in mydivs: - if 'NotImplementedError' in str(div) or 'NameError' in str(div): + if any(err in str(div) for err in ('NotImplementedError', 'NameError', 'ValueError', 'KeyboardInterrupt')): div.decompose() # Put solution figures in center (to fix layout issues)