Error was just because your inputs were not in correct order.
(I moved pi_z: Transition Paths for FHorz (minor breaking change) )
I suspect you got the order copying from some other example. If you did then let me know which I will update that example to the correct order.