Skip to content

Commit 1ec7304

Browse files
committed
fixup eval_op(ChiSq): do not repeat the formula
1 parent a979a6c commit 1ec7304

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/metrics.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ corr_dist(a::AbstractArray, b::AbstractArray) = evaluate(CorrDist(), a, b)
239239
result_type(::CorrDist, a::AbstractArray, b::AbstractArray) = result_type(CosineDist(), a, b)
240240

241241
# ChiSqDist
242-
@inline eval_op(::ChiSqDist, ai, bi) = ifelse(ai != bi, abs2(ai - bi) / (ai + bi), zero(abs2(ai - bi) / (ai + bi)))
242+
@inline eval_op(::ChiSqDist, ai, bi) = (d = abs2(ai - bi) / (ai + bi); ifelse(ai != bi, d, zero(d)))
243243
@inline eval_reduce(::ChiSqDist, s1, s2) = s1 + s2
244244
chisq_dist(a::AbstractArray, b::AbstractArray) = evaluate(ChiSqDist(), a, b)
245245

0 commit comments

Comments
 (0)