Mercurial > repos > pointsource_algs / file diff

--- a/src/forward_pdps.rs	Thu Feb 26 11:36:22 2026 -0500
+++ b/src/forward_pdps.rs	Thu Feb 26 12:55:38 2026 -0500
@@ -199,7 +199,7 @@
     // Run the algorithm
     for state in iterator.iter_init(|| full_stats(&μ, &z, ε, stats.clone())) {
         // Calculate initial transport
-        let Pair(mut τv, τz) = f.differential(Pair(&μ, &z));
+        let Pair(mut τv, τz) = f.differential(Pair(&μ, &z)) * τ;
         let μ_base = μ.clone();
 
         // Construct μ^{k+1} by solving finite-dimensional subproblems and insert new spikes.