fix: these indices were backwards

denehoffman · Oct 28, 2024 · b14a962 · b14a962
1 parent c25eb85
commit b14a962
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/likelihoods.rs b/src/likelihoods.rs
@@ -827,8 +827,8 @@ impl Function<Float, (), Infallible> for LikelihoodEvaluator {
             .zip(self.likelihood_manager.param_layouts.iter())
         {
             let term_gradient = term.evaluate_gradient(param_buffer); // This has a local layout
-            for (buffer_idx, &param_idx) in layout.iter().enumerate() {
-                gradient_buffer[buffer_idx] = term_gradient[param_idx] // This has a global layout
+            for (term_idx, &buffer_idx) in layout.iter().enumerate() {
+                gradient_buffer[buffer_idx] = term_gradient[term_idx] // This has a global layout
             }
         }
         let likelihood_gradients = LikelihoodGradients(gradient_buffers);