Eq (5) in 3.3 is inconsistent with the loss calculation in the code

[liuyueChang]

The loss calculation in the paper includes log p(Z | X) and log p(Y|Z,X), but in the code

https://github.com/roggirg/AutoBots/blob/3a61ad9f80603f99ab1e9cc535acccf72c5d6bea/utils/train_helpers.py#L72

only the p(Y|Z,X) participates in the loss calculation.

Please give me some tips, thank you!