Make display code more robust (#15)

Make display code more robust
This commit is contained in:
Eugene Yurtsev
2023-05-12 11:27:38 -04:00
committed by GitHub
parent e5330b9f9c
commit a489293c86
+6 -3
View File
@@ -42,8 +42,6 @@ class HtmlResult(TypedDict):
class _NoExpectedAnswer:
"""A sentinel class to indicate that there is no expected answer."""
pass
NO_EXPECTED_ANSWER = _NoExpectedAnswer()
@@ -61,7 +59,12 @@ def as_html_dict(
code = code_result["code"].strip()
if pretty_print:
code = AstPrinter().visit(parse(code), pretty_print=True)
try:
code = AstPrinter().visit(parse(code), pretty_print=True)
except Exception:
# This is display code. Broad exception handling OK for now
# we can make the code more robust later.
code = code
else:
code = code