From c07fabe7afb63b5f8ad3a25010ec58839d1ddf94 Mon Sep 17 00:00:00 2001 From: Riccardo Magliocchetti Date: Mon, 28 Oct 2024 13:47:51 +0100 Subject: [PATCH] please pylint --- .../tests/test_tasks.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/instrumentation/opentelemetry-instrumentation-celery/tests/test_tasks.py b/instrumentation/opentelemetry-instrumentation-celery/tests/test_tasks.py index 72117ec68b..c68b1bc758 100644 --- 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( + 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()