Move cut posterior handlers to robust module

BasisResearch · Jul 8, 2024 · 403673d · 403673d
1 parent be862d1
commit 403673d
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/chirho/observational/handlers/cut.py → chirho/robust/handlers/cut.py b/chirho/observational/handlers/cut.py → chirho/robust/handlers/cut.py
diff --git a/docs/source/dr_learner.ipynb b/docs/source/dr_learner.ipynb
@@ -45,7 +45,7 @@
     "import pyro.distributions as dist\n",
     "from pyro.infer.autoguide import AutoNormal\n",
     "from chirho.indexed.handlers import IndexPlatesMessenger\n",
-    "from chirho.observational.handlers.cut import SingleStageCut\n",
+    "from chirho.robust.handlers.cut import SingleStageCut\n",
     "from pyro.infer import Predictive\n",
     "\n",
     "pyro.settings.set(module_local_params=True)\n",

diff --git a/docs/source/sdid.ipynb b/docs/source/sdid.ipynb
@@ -52,7 +52,7 @@
     "import pyro.distributions as dist\n",
     "from pyro.infer.autoguide import AutoNormal\n",
     "from chirho.indexed.handlers import IndexPlatesMessenger\n",
-    "from chirho.observational.handlers.cut import SingleStageCut\n",
+    "from chirho.robust.handlers.cut import SingleStageCut\n",
     "from pyro.infer import Predictive\n",
     "\n",
     "pyro.settings.set(module_local_params=True)\n",

diff --git a/...servational/test_cut_posterior_modules.py → tests/robust/test_handlers_cut.py b/...servational/test_cut_posterior_modules.py → tests/robust/test_handlers_cut.py
@@ -7,7 +7,7 @@
 from pyro.infer.autoguide import AutoMultivariateNormal
 
 from chirho.indexed.handlers import IndexPlatesMessenger
-from chirho.observational.handlers.cut import (
+from chirho.robust.handlers.cut import (
     CutComplementModule,
     CutModule,
     SingleStageCut,