From aa8d183f4307a1de7a6e2fb676be21a4a9de9f08 Mon Sep 17 00:00:00 2001 From: Fritz Obermeyer Date: Tue, 5 Jul 2022 16:35:04 -0400 Subject: [PATCH 1/2] Expose docs for pyro.contrib.funsor.infer.infer_discrete --- docs/source/contrib.funsor.rst | 6 ++++++ pyro/ops/einsum/torch_marginal.py | 1 + 2 files changed, 7 insertions(+) diff --git a/docs/source/contrib.funsor.rst b/docs/source/contrib.funsor.rst index c0593542c2..90de151e86 100644 --- a/docs/source/contrib.funsor.rst +++ b/docs/source/contrib.funsor.rst @@ -65,3 +65,9 @@ Inference algorithms :show-inheritance: :member-order: bysource +.. automodule:: pyro.contrib.funsor.infer.discrete + :members: + :undoc-members: + :show-inheritance: + :member-order: bysource + diff --git a/pyro/ops/einsum/torch_marginal.py b/pyro/ops/einsum/torch_marginal.py index b197afbdd2..c1c1ce1dfc 100644 --- a/pyro/ops/einsum/torch_marginal.py +++ b/pyro/ops/einsum/torch_marginal.py @@ -54,6 +54,7 @@ def einsum(equation, *operands): if any(hasattr(x, "_pyro_backward") for x in operands): result._pyro_backward = _EinsumBackward(equation, operands) + import pdb; pdb.set_trace() return result From 7fb0266a99433d8453375382553ba72870b4a2ee Mon Sep 17 00:00:00 2001 From: Fritz Obermeyer Date: Tue, 5 Jul 2022 16:35:56 -0400 Subject: [PATCH 2/2] Revert accidental change --- pyro/ops/einsum/torch_marginal.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pyro/ops/einsum/torch_marginal.py b/pyro/ops/einsum/torch_marginal.py index c1c1ce1dfc..b197afbdd2 100644 --- a/pyro/ops/einsum/torch_marginal.py +++ b/pyro/ops/einsum/torch_marginal.py @@ -54,7 +54,6 @@ def einsum(equation, *operands): if any(hasattr(x, "_pyro_backward") for x in operands): result._pyro_backward = _EinsumBackward(equation, operands) - import pdb; pdb.set_trace() return result