feat(PassOn): a simple task that immediately passes its input

This is used for connecting tasks into slightly difficult ways.
radiocosmology · Dec 21, 2023 · 548cb07 · 548cb07
1 parent 58efa6a
commit 548cb07
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/draco/core/misc.py b/draco/core/misc.py
@@ -298,3 +298,15 @@ def setup(self, input_):
     def next(self, input_):
         """Immediately forward any input."""
         return input_
+
+
+class PassOn(task.MPILoggedTask):
+    """Unconditionally forward a tasks input.
+
+    While this seems like a pointless no-op it's useful for connecting tasks in complex
+    topologies.
+    """
+
+    def next(self, input_):
+        """Immediately forward any input."""
+        return input_