From b878a7a625e51b5283c6fd36fe2984f53994b3e5 Mon Sep 17 00:00:00 2001 From: Adam Li Date: Sun, 25 Aug 2024 05:34:38 -0400 Subject: [PATCH] Fix multiscale core to not require class conditional (#65) Signed-off-by: Adam Li --- normflows/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/normflows/core.py b/normflows/core.py index b4b0c88..9920193 100644 --- a/normflows/core.py +++ b/normflows/core.py @@ -585,12 +585,12 @@ def sample(self, num_samples=1, y=None, temperature=None): self.reset_temperature() return z, log_q - def log_prob(self, x, y): + def log_prob(self, x, y=None): """Get log probability for batch Args: x: Batch - y: Classes of x + y: Classes of x. Must be passed in if `class_cond` is True. Returns: log probability