Remove for loop around the abs function now the abs function has been…

… vectorised since 2.30
stan-dev · Sep 26, 2024 · 19e6a5e · 19e6a5e
1 parent 36f8c73
commit 19e6a5e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/stan_files/lm.stan b/src/stan_files/lm.stan
@@ -105,11 +105,10 @@ model {
     if (K > 1) 
       target += beta_lpdf(R2 | half_K, eta);
     else {
-      // TODO(Andrew) remove once vectorised abs available in rstan
+
       array[J] real R2_abs;
-      for (j in 1:J) {
-        R2_abs[j] = abs(R2[j]);
-      }
+      R2_abs = abs(R2);
+
       target += beta_lpdf(square(R2) | half_K, eta) + sum(log(R2_abs));
     }
   }