Fix multiscale core to not require class conditional (#65)

Signed-off-by: Adam Li <[email protected]>
VincentStimper · Aug 25, 2024 · b878a7a · b878a7a
1 parent 848277e
commit b878a7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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