Amend ExceptionWithTraceback import

open-telemetry · Sep 1, 2023 · fb909ed · fb909ed
1 parent d09933a
commit fb909ed
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/...opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/__init__.py b/...opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/__init__.py
@@ -76,8 +76,13 @@ def add(x, y):
 from opentelemetry.propagators.textmap import Getter
 from opentelemetry.semconv.trace import SpanAttributes
 from opentelemetry.trace.status import Status, StatusCode
+from billiard import VERSION
 
-ExceptionWithTraceback = utils.import_exception_with_traceback()
+
+if VERSION >= (4, 0, 1):
+    from billiard.einfo import ExceptionWithTraceback
+else:
+    ExceptionWithTraceback = None
 
 logger = logging.getLogger(__name__)
 

diff --git a/...on/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/utils.py b/...on/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"""