feat: export plot_density in the public API

Author: RussellSB

pathmc/__init__.py

@@ -40,6 +40,7 @@
 from pathmc.interpret import InterpretResult, datagrid  # noqa: E402
 from pathmc._model import PathModel, model, simulate  # noqa: E402
 from pathmc.adjustment import AdjustmentModel  # noqa: E402
+from pathmc.plotting import plot_density  # noqa: E402
 from pathmc.refute import PlaceboRefutationResult  # noqa: E402
 from pathmc.sensitivity import SensitivityResult  # noqa: E402
 from pathmc.simulate import DoResult, EstimandResult  # noqa: E402
@@ -70,6 +71,7 @@
     "dag_to_spec",
     "datagrid",
     "model",
+    "plot_density",
     "register_transform",
     "same_markov_equivalence_class",
     "simulate",

tests/test_public_api.py

@@ -42,6 +42,7 @@ def test_top_level_public_api_is_explicit() -> None:
         "dag_to_spec",
         "datagrid",
         "model",
+        "plot_density",
         "register_transform",
         "same_markov_equivalence_class",
         "simulate",
@@ -64,6 +65,7 @@ def test_public_submodule_exports_are_intentional() -> None:
         "pathmc.introspect": [],
         "pathmc.panel": ["PanelInfo"],
         "pathmc.parse": [],
+        "pathmc.plotting": ["plot_density"],
         "pathmc.priors": [],
         "pathmc.refute": ["PlaceboRefutationResult", "refute_placebo"],
         "pathmc.residuals": [],