Update

winter-telescope · Dec 6, 2024 · 1f49b82 · 1f49b82
1 parent 24543e7
commit 1f49b82
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mirar/pipelines/wasp/blocks.py b/mirar/pipelines/wasp/blocks.py
@@ -64,7 +64,9 @@
 load_raw = [
     ImageLoader(input_sub_dir="raw", load_image=load_raw_wasp_image),
     CustomImageBatchModifier(label_stack_id),
-    ImageBatcher(BASE_NAME_KEY),
+    ImageRebatcher(split_key="stackid"),
+    CustomImageBatchModifier(annotate_target_coordinates),
+    ImageRebatcher(BASE_NAME_KEY),
 ]
 
 
@@ -105,7 +107,6 @@
         cache=False,
     ),
     ImageRebatcher(split_key=["stackid"]),
-    CustomImageBatchModifier(annotate_target_coordinates),
     Swarp(
         swarp_config_path=swarp_config_path,
         include_scamp=True,