Skip to content

Commit 2723691

Browse files
Merge pull request #407 from OP-TED/feature/TED-1080
Update DagBatchPipelineOperator.py
2 parents 1da0f61 + a38012b commit 2723691

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

dags/operators/DagBatchPipelineOperator.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
smart_xcom_push
99
from dags.pipelines.pipeline_protocols import NoticePipelineCallable
1010
from ted_sws import config
11-
from ted_sws.core.model.notice import NoticeStatus
1211
from ted_sws.data_manager.adapters.notice_repository import NoticeRepository
1312
from ted_sws.event_manager.model.event_message import EventMessage, NoticeEventMessage
1413
from ted_sws.event_manager.services.log import log_notice_error
@@ -18,7 +17,7 @@
1817
START_WITH_STEP_NAME_KEY = "start_with_step_name"
1918
EXECUTE_ONLY_ONE_STEP_KEY = "execute_only_one_step"
2019
DEFAULT_NUBER_OF_CELERY_WORKERS = 144 # TODO: revise this config
21-
NOTICE_PROCESS_WORKFLOW_DAG_NAME = "notice_process_workflow"
20+
NOTICE_PROCESSING_PIPELINE_DAG_NAME = "notice_processing_pipeline"
2221
DEFAULT_START_WITH_TASK_ID = "notice_normalisation_pipeline"
2322
DEFAULT_PIPELINE_NAME_FOR_LOGS = "unknown_pipeline_name"
2423

@@ -136,7 +135,7 @@ def execute(self, context: Any):
136135
for notice_batch in chunks(notice_ids, chunk_size=batch_size):
137136
TriggerDagRunOperator(
138137
task_id=f'trigger_worker_dag_{uuid4().hex}',
139-
trigger_dag_id=NOTICE_PROCESS_WORKFLOW_DAG_NAME,
138+
trigger_dag_id=NOTICE_PROCESSING_PIPELINE_DAG_NAME,
140139
conf={
141140
NOTICE_IDS_KEY: list(notice_batch),
142141
START_WITH_STEP_NAME_KEY: self.start_with_step_name,

0 commit comments

Comments
 (0)