fix entropy term in distributions.py (wrong isinstance check) (#3120)

pyro-ppl · Jul 28, 2022 · 0e0b1ed · 0e0b1ed
1 parent cdea256
commit 0e0b1ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/distributions/torch_distribution.py b/pyro/distributions/torch_distribution.py
@@ -479,7 +479,7 @@ def score_parts(self, value):
             log_prob = log_prob.expand(shape)
             if isinstance(score_function, torch.Tensor):
                 score_function = score_function.expand(shape)
-            if isinstance(score_function, torch.Tensor):
+            if isinstance(entropy_term, torch.Tensor):
                 entropy_term = entropy_term.expand(shape)
         return ScoreParts(log_prob, score_function, entropy_term)