Fix type issue in FW by promoting to float

JuliaDecisionFocusedLearning · Jul 13, 2022 · 256d64a · 256d64a
1 parent b179f16
commit 256d64a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)