Update fit.jl (#175)

* Update fit.jl

correct r2 with weight + save residuals

* Update fit.jl

* Update predict.jl

* Update Project.toml

Author: matthieugomez

Project.toml

@@ -1,6 +1,6 @@
 name = "FixedEffectModels"
 uuid = "9d5cd8c9-2029-5cab-9928-427838db53e3"
-version = "1.6.1"
+version = "1.6.2"
 
 [deps]
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"

src/fit.jl

@@ -337,9 +337,10 @@ function reg(
     if save_residuals
         residuals2 = Vector{Union{Float64, Missing}}(missing, N)
         if has_weights
-            residuals .= residuals ./ sqrt.(weights)
+            residuals2[esample] .= residuals ./ sqrt.(weights)
+        else
+            residuals2[esample] .= residuals
         end
-        residuals2[esample] = residuals
     end
 
     augmentdf = DataFrame()