Merge pull request #21 from axelparmentier/giom

Fix type issue in FW by promoting to float
JuliaDecisionFocusedLearning · Jul 13, 2022 · 5062fdc · 5062fdc
2 parents 693b634 + 256d64a
commit 5062fdc
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)