Plot improvements (#598)

* wip * bring plot_func_model in shape * some stylistic improvement to two choice version * n choices functional now * lots of fixes, adding in tests * drop stray files * fix tests * stray print * fix wrong tuple return * address comments and improve tests a bit * address comments and improve tests a bit * typing insanity * drop Dict import * fix indentation * address remaining comments * more mypy pain
lnccbrown · Nov 28, 2024 · 581c0fd · 581c0fd
1 parent 6713546
commit 581c0fd
Show file tree

Hide file tree

Showing 12 changed files with 5,383 additions and 50 deletions.
diff --git a/src/hssm/plotting/__init__.py b/src/hssm/plotting/__init__.py
@@ -1,6 +1,11 @@
 """Plotting functionalities for HSSM."""
 
+from .model_cartoon import plot_model_cartoon
 from .posterior_predictive import plot_posterior_predictive
 from .quantile_probability import plot_quantile_probability
 
-__all__ = ["plot_posterior_predictive", "plot_quantile_probability"]
+__all__ = [
+    "plot_posterior_predictive",
+    "plot_quantile_probability",
+    "plot_model_cartoon",
+]