another type restriction

JuliaStats · Jun 24, 2024 · 2a190e3 · 2a190e3
1 parent 1bab6e7
commit 2a190e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/linearmixedmodel.jl b/src/linearmixedmodel.jl
@@ -1339,7 +1339,7 @@ function varest(m::LinearMixedModel)
     return isnothing(m.optsum.sigma) ? pwrss(m) / ssqdenom(m) : m.optsum.sigma
 end
 
-function StatsAPI.weights(m::LinearMixedModel)
+function StatsAPI.weights(m::LinearMixedModel{T}) where {T}
     rtwts = m.sqrtwts
-    return isempty(rtwts) ? ones(eltype(rtwts), nobs(m)) : abs2.(rtwts)
+    return isempty(rtwts) ? ones(T, nobs(m)) : abs2.(rtwts)
 end