Update format.py (#23)

McGill-NLP · Apr 4, 2024 · d00cc3f · d00cc3f
1 parent 6d24d84
commit d00cc3f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/weblinx/utils/format.py b/weblinx/utils/format.py
@@ -468,6 +468,8 @@ def format_uid(turn, uid_key="data-webtasks-id", return_as="dict"):
     formatted : str or dict
         A string or dictionary representing the uid.
     """
+    if turn.element is None:
+        raise ValueError(f"format_uid received a turn object with turn.element missing (None): {turn}")
     output = {"uid": turn.element.get("attributes", {}).get(uid_key, None)}
 
     return format_output_dictionary(output, return_as=return_as)