please pylint

open-telemetry · Oct 28, 2024 · 9b16dcd · 9b16dcd
1 parent 73befba
commit 9b16dcd
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/instrumentation/opentelemetry-instrumentation-celery/tests/test_tasks.py b/instrumentation/opentelemetry-instrumentation-celery/tests/test_tasks.py
@@ -188,20 +188,20 @@ def test_baggage(self):
 
         self.assertEqual(task.result, {"key": "value"})
 
-    def _retrieve_context_wrapper_none_token(
-        self, wrapped, instance, args, kwargs
-    ):
-        ctx = wrapped(*args, **kwargs)
-        if ctx is None:
-            return ctx
-        span, activation, _ = ctx
-        return span, activation, None
-
     def test_task_not_instrumented_does_not_raise(self):
+        def _retrieve_context_wrapper_none_token(
+            self, wrapped, instance, args, kwargs
+        ):
+            ctx = wrapped(*args, **kwargs)
+            if ctx is None:
+                return ctx
+            span, activation, _ = ctx
+            return span, activation, None
+
         wrap_function_wrapper(
             utils,
             "retrieve_context",
-            self._retrieve_context_wrapper_none_token,
+            _retrieve_context_wrapper_none_token,
         )
 
         CeleryInstrumentor().instrument()