@@ -79,8 +79,8 @@ tag f = khatri f identity
7979
8080state g s r = tag g `comp` tag s `comp` r
8181
82−grass_wet :: Matrix Double (G, (S, R)) One
83−grass_wet = row [0,1] . kp1
82+grass_wet :: Matrix Double () (G, (S, R)) -> Matrix Double One One
83+grass_wet state = row [0,1] . kp1 . state
8484
8585rainning :: Matrix Double (G, (S, R)) One
8686rainning = row [0,1] . kp2 . kp2
@@ -162,7 +162,7 @@ exec = do
162162 putStrLn "\n Checking that the last result is indeed a distribution: "
163163 prettyPrint (bang . sumSSM . khatri die die)
164164 putStrLn "\n Probability of grass being wet:"
165− prettyPrint (grass_wet . state grass sprinkler rain)
165+ prettyPrint (grass_wet (state grass sprinkler rain))
166166 putStrLn "\n Probability of rain:"
167167 prettyPrint (rainning . state grass sprinkler rain)
168168 putStrLn "\n Is the arbitrary state a valid state? (Alcuin Puzzle)"