From da6f333ab0cec474ad89c99228a62129274117bc Mon Sep 17 00:00:00 2001
From: Paul Xu <yang_xu@brown.edu>
Date: Tue, 12 Dec 2023 15:50:33 -0500
Subject: [PATCH] added quantile probability plot as a method for hssm

---
 src/hssm/hssm.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/hssm/hssm.py b/src/hssm/hssm.py
index d17b2782..c716f136 100644
--- a/src/hssm/hssm.py
+++ b/src/hssm/hssm.py
@@ -496,6 +496,20 @@ def plot_posterior_predictive(self, **kwargs) -> mpl.axes.Axes | sns.FacetGrid:
         """
         return plotting.plot_posterior_predictive(self, **kwargs)
 
+    def plot_quantile_probability(self, **kwargs) -> mpl.axes.Axes | sns.FacetGrid:
+        """Produce a quantile probability plot.
+
+        Equivalent to calling `hssm.plotting.plot_quantile_probability()` with the
+        model. Please see that function for
+        [full documentation][hssm.plotting.plot_quantile_probability].
+
+        Returns
+        -------
+        mpl.axes.Axes | sns.FacetGrid
+            The matplotlib axis or seaborn FacetGrid object containing the plot.
+        """
+        return plotting.plot_quantile_probability(self, **kwargs)
+
     def sample_prior_predictive(
         self,
         draws: int = 500,