Skip to content

Commit

Permalink
Merge branch 'main' into staging/fix-crawler-ans
Browse files Browse the repository at this point in the history
  • Loading branch information
tricktx authored Aug 14, 2024
2 parents 767f937 + 4fbcbbb commit 522e999
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion pipelines/datasets/br_rf_cno/flows.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,7 @@
dump_mode=unmapped("append"),
source_format=unmapped('parquet'),
wait=unmapped(files),
upstream_tasks=[files]
)

dbt_parameters = create_parameters_list(
Expand All @@ -107,6 +108,7 @@
download_csv_file = True,
dbt_command = 'run',
disable_elementary = True,
upstream_tasks=[wait_upload_table]
)

with case(materialize_after_dump, True):
Expand All @@ -118,13 +120,15 @@
parameters=dbt_parameters,
labels=unmapped(current_flow_labels),
run_name=f"Materialize {dataset_id}.{table_ids}",
upstream_tasks=[dbt_parameters]
)

wait_for_materialization = wait_for_flow_run.map(
materialization_flow,
stream_states=unmapped(True),
stream_logs=unmapped(True),
raise_final_state=unmapped(True),
upstream_tasks=[materialization_flow]
)
wait_for_materialization.max_retries = (
dump_db_constants.WAIT_FOR_MATERIALIZATION_RETRY_ATTEMPTS.value
Expand All @@ -137,7 +141,7 @@
update_django_metadata.map(
dataset_id=unmapped(dataset_id),
table_id=table_ids,
date_column_name=unmapped({"date": "data"}),
date_column_name=unmapped({"date": "data_extracao"}),
date_format=unmapped("%Y-%m-%d"),
coverage_type=unmapped("all_bdpro"),
#time_delta=unmapped({"months": 6}),
Expand Down

0 comments on commit 522e999

Please sign in to comment.