change case to render graph (pyro-ppl#1711)

aleatory-science · May 6, 2024 · afabd2b · afabd2b
1 parent e5a18ed
commit afabd2b
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"
     }