grouped ping support for image builder

themachinethatgoesping · May 14, 2024 · b52e704 · b52e704
1 parent d1cd7d3
commit b52e704
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/python/themachinethatgoesping/pingprocessing/watercolumn/image/imagebuilder.py b/python/themachinethatgoesping/pingprocessing/watercolumn/image/imagebuilder.py
@@ -21,6 +21,11 @@ def __init__(
         self.use_beam_sample_view = use_beam_sample_view
         self.progress = progress
 
+        # if isinstance(self.pings, dict):
+        #     self.dual_head = True
+        # else:
+        #     self.dual_head = False
+
     def update_args(self, **kwargs):
         self.default_args.update(kwargs)
 
@@ -46,10 +51,14 @@ def build(self, index, stack = 1, stack_step = 1, **kwargs):
                 progress=self.progress,
                 **_kwargs)
 
-        return make_wci(
+        return make_wci_dual_head(
             self.pings[index],
             **_kwargs)
 
+        # return make_wci(
+        #     self.pings[index],
+        #     **_kwargs)
+
 
 
 # progress_bar.description = "Idle"