From 84ca7f07cb1248ff097d977f7827be92d9029250 Mon Sep 17 00:00:00 2001
From: Oscar Dowson <odow@users.noreply.github.com>
Date: Thu, 28 Mar 2024 19:24:49 +1300
Subject: [PATCH] Fix _init_identity_matrix in bfgs.jl

---
 src/multivariate/solvers/first_order/bfgs.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/multivariate/solvers/first_order/bfgs.jl b/src/multivariate/solvers/first_order/bfgs.jl
index 14635ab43..32a465d40 100644
--- a/src/multivariate/solvers/first_order/bfgs.jl
+++ b/src/multivariate/solvers/first_order/bfgs.jl
@@ -58,11 +58,11 @@ mutable struct BFGSState{Tx, Tm, T,G} <: AbstractOptimizerState
     @add_linesearch_fields()
 end
 
-function _init_identity_matrix(x::AbstractArray{T}, scale::T = T(1)) where {T}
+function _init_identity_matrix(x::AbstractArray{T}, scale = T(1)) where {T}
     x_ = reshape(x, :)
     Id = x_ .* x_' .* false
     idxs = diagind(Id)
-    @. @view(Id[idxs]) = scale * true
+    @. @view(Id[idxs]) = T(scale) * true
     return Id
 end