From fb909ed000c74f2f67a22457bf5c94c7c25eb72c Mon Sep 17 00:00:00 2001
From: Noemi <45180344+unflxw@users.noreply.github.com>
Date: Fri, 1 Sep 2023 17:31:04 +0200
Subject: [PATCH] Amend ExceptionWithTraceback import

---
 .../src/opentelemetry/instrumentation/celery/__init__.py | 7 ++++++-
 .../src/opentelemetry/instrumentation/celery/utils.py    | 9 ---------
 2 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/__init__.py b/instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/__init__.py
index d4e53ff875..bb83a5c192 100644
--- a/instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/__init__.py
+++ b/instrumentation/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/instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/utils.py b/instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/utils.py
index acf949d551..f92c5e03c8 100644
--- a/instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/utils.py
+++ b/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"""