Shorten comment

pyro-ppl · Dec 18, 2023 · fc5c5c0 · fc5c5c0
1 parent 96cfaf6
commit fc5c5c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/contrib/zuko.py b/pyro/contrib/zuko.py
@@ -27,7 +27,7 @@ def batch_shape(self) -> Size:
         return self.dist.batch_shape
 
     def __call__(self, shape: Size = ()) -> Tensor:
-        if hasattr(self.dist, "rsample_and_log_prob"):  # special method for fast sampling + scoring
+        if hasattr(self.dist, "rsample_and_log_prob"):  # fast sampling + scoring
             x, self.cache[x] = self.dist.rsample_and_log_prob(shape)
         elif self.has_rsample:
             x = self.dist.rsample(shape)