change case to render graph (#1711)

pyro-ppl · Dec 30, 2023 · 9ce2384 · 9ce2384
1 parent 09551d9
commit 9ce2384
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro/infer/inspect.py b/numpyro/infer/inspect.py
@@ -385,7 +385,7 @@ def process_message(self, msg):
     samples = {
         name: site["value"]
         for name, site in trace.items()
-        if (site["type"] == "sample" and not site["is_observed"])
+        if site["type"] == "sample"
         or site["type"] == "deterministic"
     }