diff --git a/pyro/infer/inspect.py b/pyro/infer/inspect.py index 2580301dba..18055bba38 100644 --- a/pyro/infer/inspect.py +++ b/pyro/infer/inspect.py @@ -298,7 +298,7 @@ def model(data): assert isinstance(model_kwargs, dict) with torch.random.fork_rng(), torch.no_grad(), pyro.validation_enabled(False): - with TrackProvenance(include_deterministic=include_deterministic): + with TrackProvenance(include_deterministic=True): trace = poutine.trace(model).get_trace(*model_args, **model_kwargs) sample_sample = {} diff --git a/tests/infer/test_inspect.py b/tests/infer/test_inspect.py index 8b72c621ca..baccf00500 100644 --- a/tests/infer/test_inspect.py +++ b/tests/infer/test_inspect.py @@ -526,7 +526,7 @@ def model(data): "f": ["e"], "g": ["e"], "h": ["e"], - "i": ["e"], + "i": ["e", "f", "g", "h"], }, "sample_param": { "a": [],