In error_utils, instance of current_context_msg should be shared across all threads/coroutines.

The TensorFlow Datasets Authors · The TensorFlow Datasets Authors · commit d9c01103a307 · 2024-07-01T06:26:50.000-07:00
PiperOrigin-RevId: 648347714
diff --git a/tensorflow_datasets/core/utils/error_utils.py b/tensorflow_datasets/core/utils/error_utils.py
@@ -41,7 +41,7 @@ class ErrorContext:
 @edc.dataclass
 @dataclasses.dataclass
 class ContextHolder:
-  current_context_msg: edc.ContextVar[ErrorContext | None] = None
+  current_context_msg: ErrorContext | None = None
 
 
 context_holder = ContextHolder()
