bugfix: setProbabilities should be more robust when there are missing probabilities for some functions or categories

2026-04-09 04:59:31 -06:00 · 2010-12-17 09:09:07 +00:00
parent 2057c9f284
commit e4b2b8541f
1 changed files with 1 additions and 1 deletions
--- a/src/runtime/haskell/PGF/Probabilistic.hs
+++ b/src/runtime/haskell/PGF/Probabilistic.hs
@@ -80,7 +80,7 @@ setProbabilities probs pgf = pgf {
  }}
  where
    mapUnionWith f map1 map2 = 
-      Map.mapWithKey (\k v -> f v (fromJust (Map.lookup k map2))) map1
+      Map.mapWithKey (\k v -> maybe v (f v) (Map.lookup k map2)) map1

 -- | compute the probability of a given tree
 probTree :: PGF -> Expr -> Double