TMP - modified

project-oak · Mar 4, 2022 · bdb7725 · bdb7725
1 parent 7244d0c
commit bdb7725
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ibis/src/recipes.rs b/ibis/src/recipes.rs
@@ -104,10 +104,11 @@ ibis! {
         KnownType(apply!(y_generic, y_arg));
 
     HasTag(s, n, n, tag) <- Solution(s), Claim(n, tag);
-    HasTag(s, source, *down, tag) <- // Propagate tags 'downstream'
+    HasTag(s, source, down, tag) <- // Propagate tags 'downstream'
+        Node(_down_particle, down, _, _),
         HasTag(s, source, curr, tag),
-        for (curr, down) in &s.solution().edges,
-        !TrustedToRemoveTag(*down, tag);
+        !TrustedToRemoveTag(down, tag),
+        (s.has_edge(curr, down));
 
     HasTag(s, source, down, tag) <- // Propagate tags 'across stream' (i.e. inside a particle)
         HasTag(s, source, curr, tag),