Update EqualizeMessenger docs.

pyro-ppl · Jun 16, 2024 · 03fcce4 · 03fcce4
1 parent 0d1b69c
commit 03fcce4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyro/poutine/equalize_messenger.py b/pyro/poutine/equalize_messenger.py
@@ -13,9 +13,9 @@
 
 class EqualizeMessenger(Messenger):
     """
-    Given a stochastic function with some primitive statements
-    and a list of names,
-    force the primitive statements at those names to have the same value.
+    Given a stochastic function with some primitive statements and a list of names,
+    force the primitive statements at those names to have the same value,
+    with that value being the result of the first primitive statement matching those names.
 
     Consider the following Pyro program: