Hi shi_le, the TFP github repo code should generally be the same as tfp-nightly, modulo changes within the last day. It’s possible you had a few days old snapshot of the TFP repo, since I believe Emily’s recent changes fixed this (real!) bug.
If you can, please try with tfp-nightly from pip, and/or rebuilt .whl from the github repo, synced to the most recent HEAD, and see if this alleviates your issue. Let us know if not!
Thanks for catching this error and discussing it with us, helping us to improve TFP. Sorry for the confusing bug!