From 383f86276a0ceb7b9733ff30610cbb10681545d6 Mon Sep 17 00:00:00 2001 From: Diego Hurtado Date: Mon, 30 Aug 2021 23:16:55 +0200 Subject: [PATCH] Fix lint --- .../tests/celery/test_celery_functional.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/opentelemetry-docker-tests/tests/celery/test_celery_functional.py b/tests/opentelemetry-docker-tests/tests/celery/test_celery_functional.py index 615c1ea24b..e38b162225 100644 --- a/tests/opentelemetry-docker-tests/tests/celery/test_celery_functional.py +++ b/tests/opentelemetry-docker-tests/tests/celery/test_celery_functional.py @@ -12,10 +12,10 @@ # See the License for the specific language governing permissions and # limitations under the License. - -import celery -from pytest import mark +from celery import shared_task from celery.exceptions import Retry +from celery.task import Task +from pytest import mark import opentelemetry.instrumentation.celery from opentelemetry import trace as trace_api @@ -451,7 +451,7 @@ def run(self): def test_shared_task(celery_app, memory_exporter): """Ensure Django Shared Task are supported""" - @celery.shared_task + @shared_task def add(x, y): return x + y @@ -484,7 +484,7 @@ def test_apply_async_previous_style_tasks( used even in newer versions. This should extend support to previous versions of Celery.""" - class CelerySuperClass(celery.task.Task): + class CelerySuperClass(Task): abstract = True @classmethod