diff --git a/python_modules/libraries/dagster-celery/dagster_celery/engine.py b/python_modules/libraries/dagster-celery/dagster_celery/engine.py index 160d89d71eca3..bdde44c297de8 100644 --- a/python_modules/libraries/dagster-celery/dagster_celery/engine.py +++ b/python_modules/libraries/dagster-celery/dagster_celery/engine.py @@ -1,8 +1,6 @@ import sys import time -from dagster_k8s.job import get_k8s_resource_requirements - from dagster import check from dagster.core.engine.engine_base import Engine from dagster.core.errors import DagsterSubprocessError @@ -189,6 +187,8 @@ def _submit_task(app, pipeline_context, step, queue, priority): def _submit_task_k8s_job(app, pipeline_context, step, queue, priority): from .tasks import create_k8s_job_task + from dagster_k8s.job import get_k8s_resource_requirements + resources = get_k8s_resource_requirements(step.tags) task = create_k8s_job_task(app)