fix: change integrand check order in KL divergence code

2022-05-06 09:49:42 -04:00 · 2022-05-06 09:49:42 -04:00 · 87d6f8d4f6
commit 87d6f8d4f6
parent fd17da83b8
1 changed files with 3 additions and 3 deletions
--- a/packages/squiggle-lang/src/rescript/Distributions/PointSetDist/PointSetDist_Scoring.res
+++ b/packages/squiggle-lang/src/rescript/Distributions/PointSetDist/PointSetDist_Scoring.res
@ -4,10 +4,10 @@ module KLDivergence = {
    float,
    Operation.Error.t,
  > =>
-    if predictionElement == 0.0 {
-      Error(Operation.NegativeInfinityError)
-    } else if answerElement == 0.0 {
+    if answerElement == 0.0 {
      Ok(0.0)
+    } else if predictionElement == 0.0 {
+      Error(Operation.NegativeInfinityError)
    } else {
      let quot = predictionElement /. answerElement
      quot < 0.0 ? Error(Operation.ComplexNumberError) : Ok(answerElement *. logFn(quot))