ModuleNotFoundError: No module named 'flowMC.nfmodel'

[dirknbr]

getting this error when I run in colab

3 import tensorflow_probability as tfp
      4 tfd = tfp.distributions
----> 5 import bayeux as bx
      6 import jax

2 frames
[/usr/local/lib/python3.11/dist-packages/bayeux/__init__.py](https://localhost:8080/#) in <module>
     22 # See PEP 484 & https://github.com/google/jax/issues/7570
     23 # pylint: disable=useless-import-alias
---> 24 from bayeux import mcmc as mcmc
     25 from bayeux import optimize as optimize
     26 from bayeux._src import debug as debug

[/usr/local/lib/python3.11/dist-packages/bayeux/mcmc/__init__.py](https://localhost:8080/#) in <module>
     37 
     38 if importlib.util.find_spec("flowMC") is not None:
---> 39   from bayeux._src.mcmc.flowmc import MaskedCouplingRQSplineHMC as MaskedCouplingRQSplineHMCflowmc
     40   from bayeux._src.mcmc.flowmc import MaskedCouplingRQSplineMALA as MaskedCouplingRQSplineMALAflowmc
     41   from bayeux._src.mcmc.flowmc import RealNVPHMC as RealNVPHMCflowmc

[/usr/local/lib/python3.11/dist-packages/bayeux/_src/mcmc/flowmc.py](https://localhost:8080/#) in <module>
     31 from bayeux._src import shared
     32 from flowMC import Sampler
---> 33 from flowMC.nfmodel import realNVP
     34 from flowMC.nfmodel import rqSpline
     35 from flowMC.proposal import HMC

ModuleNotFoundError: No module named 'flowMC.nfmodel'

[dcstats]

@dirknbr seems like an issue with flowMC>=0.4.0, I got around the issue by installing flowMC==0.3.4