From a0acdc794dd77cb7915fa7cdc76104cd4930cb26 Mon Sep 17 00:00:00 2001
From: Joakim Skogholt <onlinereg@skogholt.xyz>
Date: Thu, 25 Apr 2024 18:12:25 +0200
Subject: [PATCH] n-th time the charm

---
 src/TR.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/TR.jl b/src/TR.jl
index 36fbe64..2a0ac86 100644
--- a/src/TR.jl
+++ b/src/TR.jl
@@ -484,8 +484,8 @@ for j = 1:length(lambdas)
         Xs = Xdata .- mX;
         ys = ydata .- my;    
         
-        betas         = [Xs; sqrt(lambdas[j]) * I(p)] \ [ys; zeros(p,1)];
-        rmsecvman[j] += sum((y[vec(inds)] - ((X[vec(inds),:] .- mX) * betas .+ my)).^2);
+        betas      = [Xs; sqrt(lambdas[j]) * I(p)] \ [ys; zeros(p,1)];
+        rmsecv[j] += sum((y[vec(inds)] - ((X[vec(inds),:] .- mX) * betas .+ my)).^2);
     end
 end