From e05cb06b3b293409750bd0cb929bdae6a4daf138 Mon Sep 17 00:00:00 2001
From: yayami <116920988+yayami3@users.noreply.github.com>
Date: Fri, 29 Dec 2023 02:44:24 +0000
Subject: [PATCH] change case to render graph

---
 numpyro/infer/inspect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/numpyro/infer/inspect.py b/numpyro/infer/inspect.py
index 2b97f4021..5232dfe11 100644
--- 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"
     }