added zip to archive futures

tjgalvin · Dec 29, 2024 · 4fcdef8 · 4fcdef8
1 parent 06a9e7e
commit 4fcdef8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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(