Move cut posterior handlers to robust module (#549)

* Move cut posterior handlers to robust module * Update test_handlers_cut.py
BasisResearch · Jul 16, 2024 · 8ba2f2c · 8ba2f2c
1 parent 730eb9b
commit 8ba2f2c
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 7 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,11 +7,7 @@
 from pyro.infer.autoguide import AutoMultivariateNormal
 
 from chirho.indexed.handlers import IndexPlatesMessenger
-from chirho.observational.handlers.cut import (
-    CutComplementModule,
-    CutModule,
-    SingleStageCut,
-)
+from chirho.robust.handlers.cut import CutComplementModule, CutModule, SingleStageCut
 
 pyro.settings.set(module_local_params=True)