Add mode function to Continuous class (#650)

arviz-devs · Feb 22, 2025 · d7ed83a · d7ed83a
1 parent 9bd36b9
commit d7ed83a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/preliz/distributions/distributions.py b/preliz/distributions/distributions.py
@@ -11,7 +11,7 @@
 import numpy as np
 
 from preliz.internal.distribution_helper import init_vals, valid_distribution, valid_scalar_params
-from preliz.internal.optimization import optimize_hdi
+from preliz.internal.optimization import find_mode, optimize_hdi
 from preliz.internal.plot_helper import (
     check_inside_notebook,
     get_slider,
@@ -795,6 +795,9 @@ def __init__(self):
         super().__init__()
         self.kind = "continuous"
 
+    def mode(self):
+        return find_mode(self)
+
 
 class Discrete(Distribution):
     """Base class for discrete distributions."""

diff --git a/preliz/distributions/rice.py b/preliz/distributions/rice.py
@@ -3,7 +3,7 @@
 
 from preliz.distributions.distributions import Continuous
 from preliz.internal.distribution_helper import all_not_none, eps
-from preliz.internal.optimization import find_mode, optimize_ml, optimize_moments_rice
+from preliz.internal.optimization import optimize_ml, optimize_moments_rice
 from preliz.internal.special import cdf_bounds, ppf_bounds_cont
 
 
@@ -129,9 +129,6 @@ def entropy(self):
     def mean(self):
         return self.sigma * np.sqrt(np.pi / 2) * _l_half(-(self.nu**2) / (2 * self.sigma**2))
 
-    def mode(self):
-        return find_mode(self)
-
     def median(self):
         return self.ppf(0.5)