From 4fcdef845d9290a45c568d8c4aa72ce9d9d35829 Mon Sep 17 00:00:00 2001 From: tgalvin Date: Sun, 29 Dec 2024 18:00:54 +1100 Subject: [PATCH] added zip to archive futures --- flint/prefect/flows/continuum_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/flint/prefect/flows/continuum_pipeline.py b/flint/prefect/flows/continuum_pipeline.py index 73936089..ab1c1fd1 100644 --- a/flint/prefect/flows/continuum_pipeline.py +++ b/flint/prefect/flows/continuum_pipeline.py @@ -485,7 +485,8 @@ def process_science_fields( # zip up the final measurement set, which is not included in the above loop if field_options.zip_ms: - task_zip_ms.map(in_item=wsclean_cmds, wait_for=archive_wait_for) + zip_futures = task_zip_ms.map(in_item=wsclean_cmds, wait_for=archive_wait_for) + archive_wait_for.extend(zip_futures) if field_options.sbid_archive_path or field_options.sbid_copy_path: update_archive_options = get_options_from_strategy(