Checks for multioutput plugins

AxFoundation · Sep 2, 2020 · a19cc21 · a19cc21
1 parent e94b1c7
commit a19cc21
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/strax/plugin.py b/strax/plugin.py
@@ -608,6 +608,9 @@ def compute(self, **kwargs):
             loop_over = self.loop_over
         else:
             loop_over = self.deps[self.depends_on[0]].data_kind
+            if isinstance(loop_over, dict):
+                loop_over = loop_over[self.depends_on[0]]
+                # We can't have nested dictionaries here, can we?
 
         # Group into lists of things (e.g. peaks)
         # contained in the base things (e.g. events)