Use reraise instead of add_note in naming.py and split_builder.py.

PiperOrigin-RevId: 651504582

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/naming.py

@@ -26,6 +26,7 @@
 
 from absl import logging
 from etils import epath
+from tensorflow_datasets.core.utils import py_utils
 from tensorflow_datasets.core.utils import version as version_lib
 
 _NAME_CLASS = r'[a-zA-Z][\w]*'
@@ -185,8 +186,7 @@ def _dataset_name_and_kwargs_from_name_str(
       kwargs[attr] = val
     return name, kwargs
   except Exception as e:  # pylint: disable=broad-except
-    e.add_note(err_msg)
-    raise e
+    py_utils.reraise(e, prefix=err_msg)  # pytype: disable=bad-return-type
 
 
 @dataclasses.dataclass(order=True)

tensorflow_datasets/core/split_builder.py

@@ -399,14 +399,14 @@ def _build_from_generator(
       try:
         example = self._features.encode_example(example)
       except Exception as e:  # pylint: disable=broad-except
-        e.add_note(f'Failed to encode example:\n{example}\n')
-        raise
+        utils.reraise(e, prefix=f'Failed to encode example:\n{example}\n')
       writer.write(key, example)
     try:
       shard_lengths, total_size = writer.finalize()
     except Exception as e:  # pylint: disable=broad-except
-      e.add_note(f'Failed to finalize writing of split "{split_name}"')
-      raise
+      utils.reraise(
+          e, prefix=f'Failed to finalize writing of split "{split_name}"'
+      )
 
     split_info = splits_lib.SplitInfo(
         name=split_name,