Amend ExceptionWithTraceback import

Author: unflxw

instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/__init__.py

@@ -77,7 +77,11 @@ def add(x, y):
 from opentelemetry.semconv.trace import SpanAttributes
 from opentelemetry.trace.status import Status, StatusCode
 
-ExceptionWithTraceback = utils.import_exception_with_traceback()
+
+if VERSION >= (4, 0, 1):
+    from billiard.einfo import ExceptionWithTraceback
+else:
+    ExceptionWithTraceback = None
 
 logger = logging.getLogger(__name__)
 

instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/utils.py

@@ -48,15 +48,6 @@
 )
 
 
-def import_exception_with_traceback():
-    if VERSION >= (4, 0, 1):
-        from billiard.einfo import ExceptionWithTraceback
-
-        return ExceptionWithTraceback
-
-    return None
-
-
 # pylint:disable=too-many-branches
 def set_attributes_from_context(span, context):
     """Helper to extract meta values from a Celery Context"""