diff --git a/bilby/bilby_mcmc/flows.py b/bilby/bilby_mcmc/flows.py index 5fbaf196b38859660307554e8bc371ce0b03edf8..b08ea3a9386e8f69b6ec7644f896b17c287956be 100644 --- a/bilby/bilby_mcmc/flows.py +++ b/bilby/bilby_mcmc/flows.py @@ -1,17 +1,17 @@ import torch -from nflows.distributions.normal import StandardNormal -from nflows.flows.base import Flow -from nflows.nn import nets as nets -from nflows.transforms import ( +from glasflow.nflows.distributions.normal import StandardNormal +from glasflow.nflows.flows.base import Flow +from glasflow.nflows.nn import nets as nets +from glasflow.nflows.transforms import ( CompositeTransform, MaskedAffineAutoregressiveTransform, RandomPermutation, ) -from nflows.transforms.coupling import ( +from glasflow.nflows.transforms.coupling import ( AdditiveCouplingTransform, AffineCouplingTransform, ) -from nflows.transforms.normalization import BatchNorm +from glasflow.nflows.transforms.normalization import BatchNorm from torch.nn import functional as F # Turn off parallelism diff --git a/bilby/bilby_mcmc/proposals.py b/bilby/bilby_mcmc/proposals.py index 17892e050b33b89d8b391203b72140056ea29254..6100d75f89b68be8d0b0bf165cef3aaee53db5a7 100644 --- a/bilby/bilby_mcmc/proposals.py +++ b/bilby/bilby_mcmc/proposals.py @@ -754,10 +754,10 @@ class NormalizingFlowProposal(DensityEstimateProposal): @staticmethod def check_dependencies(warn=True): - if importlib.util.find_spec("nflows") is None: + if importlib.util.find_spec("glasflow") is None: if warn: logger.warning( - "Unable to utilise NormalizingFlowProposal as nflows is not installed" + "Unable to utilise NormalizingFlowProposal as glasflow is not installed" ) return False else: