ModuleNotFoundError: No module named 'tensorflow_probability'

[cookpa]

@ntustison looks like antspynet/architectures/create_normalizing_flow_model.py wants tensorflow_probability, which was removed in
#100.

[cookpa]

From the release notes for https://github.com/tensorflow/probability/releases/tag/v0.25.0

When using TensorFlow Probability with TensorFlow, you must explicitly install Keras 2 along with TensorFlow (or install tensorflow-probability[tf] or tfp-nightly[tf] to automatically install these dependencies.)

The ANTsPyNet requirements.txt now installs keras 3, so I'm not sure what to do with this.

[ntustison]

Thanks. Yeah, I'll take it out later when I get back home. It's more experimental work I'm doing with @stnava anyway.