Skip to content

Commit

Permalink
refactor: use consistent task name
Browse files Browse the repository at this point in the history
  • Loading branch information
hacherix committed Dec 23, 2024
1 parent 2601dd6 commit e8f0312
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 26 deletions.
4 changes: 2 additions & 2 deletions workflows/data_pipelines/finess/DAG.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def clean_previous_outputs():
)

@task
def finess_download_data():
def download_data():
return finess_processor.download_data()

@task
Expand All @@ -60,7 +60,7 @@ def compare_files_minio():

(
clean_previous_outputs()
>> finess_download_data()
>> download_data()
>> preprocess_finess()
>> save_date_last_modified()
>> send_file_to_minio()
Expand Down
24 changes: 12 additions & 12 deletions workflows/data_pipelines/formation/formation_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,36 +36,36 @@ def data_processing_organisme_formation():
formation_processor = FormationProcessor()

@task.bash
def formation_clean_previous_outputs():
def clean_previous_outputs():
return f"rm -rf {FORMATION_CONFIG.tmp_folder} && mkdir -p {FORMATION_CONFIG.tmp_folder}"

@task
def formation_download_data():
def download_data():
return formation_processor.download_data()

@task
def formation_preprocess_data():
def preprocess_data():
return formation_processor.preprocess_data()

@task
def formation_save_date_last_modified():
def save_date_last_modified():
return formation_processor.save_date_last_modified()

@task
def formation_send_file_to_minio():
def send_file_to_minio():
return formation_processor.send_file_to_minio()

@task
def formation_compare_files_minio():
def compare_files_minio():
return formation_processor.compare_files_minio()

(
formation_clean_previous_outputs()
>> formation_download_data()
>> formation_preprocess_data()
>> formation_save_date_last_modified()
>> formation_send_file_to_minio()
>> formation_compare_files_minio()
clean_previous_outputs()
>> download_data()
>> preprocess_data()
>> save_date_last_modified()
>> send_file_to_minio()
>> compare_files_minio()
)


Expand Down
24 changes: 12 additions & 12 deletions workflows/data_pipelines/uai/uai_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,36 +36,36 @@ def data_processing_uai():
uai_processor = UAIProcessor()

@task.bash
def uai_clean_previous_outputs():
def clean_previous_outputs():
return f"rm -rf {UAI_CONFIG.tmp_folder} && mkdir -p {UAI_CONFIG.tmp_folder}"

@task
def uai_download_data():
def download_data():
return uai_processor.download_data()

@task
def uai_preprocess_data():
def preprocess_data():
return uai_processor.preprocess_data()

@task
def uai_save_date_last_modified():
def save_date_last_modified():
return uai_processor.save_date_last_modified()

@task
def uai_send_file_to_minio():
def send_file_to_minio():
return uai_processor.send_file_to_minio()

@task
def uai_compare_files_minio():
def compare_files_minio():
return uai_processor.compare_files_minio()

(
uai_clean_previous_outputs()
>> uai_download_data()
>> uai_preprocess_data()
>> uai_save_date_last_modified()
>> uai_send_file_to_minio()
>> uai_compare_files_minio()
clean_previous_outputs()
>> download_data()
>> preprocess_data()
>> save_date_last_modified()
>> send_file_to_minio()
>> compare_files_minio()
)


Expand Down

0 comments on commit e8f0312

Please sign in to comment.