From 256d64a212c07813f4150721a0f8673056f6d404 Mon Sep 17 00:00:00 2001 From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com> Date: Wed, 13 Jul 2022 10:22:15 +0200 Subject: [PATCH] Fix type issue in FW by promoting to float --- src/frank_wolfe/differentiable_frank_wolfe.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/frank_wolfe/differentiable_frank_wolfe.jl b/src/frank_wolfe/differentiable_frank_wolfe.jl index 6978804..8aa24ac 100644 --- a/src/frank_wolfe/differentiable_frank_wolfe.jl +++ b/src/frank_wolfe/differentiable_frank_wolfe.jl @@ -93,7 +93,7 @@ function ChainRulesCore.rrule( fw_kwargs=(;), kwargs..., ) where {R1<:Real,R2<:Real} - R = promote_type(R1, R2) + R = promote_type(float(R1), float(R2)) (; linear_solver) = dfw probadist = compute_probability_distribution(dfw, θ, x0; fw_kwargs=fw_kwargs)