From df3aafce2a1e4207a86a2c30005ee8c2b924ab55 Mon Sep 17 00:00:00 2001
From: "michael.williams" <michael.williams@ligo.org>
Date: Thu, 22 Feb 2024 19:29:32 +0000
Subject: [PATCH] MAINT: use glasflow.nflows instead of nflows
---
bilby/bilby_mcmc/flows.py | 12 ++++++------
bilby/bilby_mcmc/proposals.py | 4 ++--
2 files changed, 8 insertions(+), 8 deletions(-)
diff --git a/bilby/bilby_mcmc/flows.py b/bilby/bilby_mcmc/flows.py
index 5fbaf196b..b08ea3a93 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 17892e050..6100d75f8 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:
--
GitLab