diff --git a/src/pytest_celery/api/base.py b/src/pytest_celery/api/base.py index 542e954e..6f5ca83f 100644 --- a/src/pytest_celery/api/base.py +++ b/src/pytest_celery/api/base.py @@ -236,12 +236,14 @@ def _set_nodes( tuple[CeleryTestNode]: Nodes to use for the cluster. """ return tuple( - node_cls(node) - if isinstance( - node, - CeleryTestContainer, + ( + node_cls(node) + if isinstance( + node, + CeleryTestContainer, + ) + else node ) - else node for node in nodes ) # type: ignore diff --git a/src/pytest_celery/vendors/worker/content/app.py b/src/pytest_celery/vendors/worker/content/app.py index 0cf8203d..f7ee2179 100644 --- a/src/pytest_celery/vendors/worker/content/app.py +++ b/src/pytest_celery/vendors/worker/content/app.py @@ -1,4 +1,5 @@ """Template for Celery worker application.""" + from __future__ import annotations import json