From 73976215812a2a1c4e54e1868cd156d4e08c3ba6 Mon Sep 17 00:00:00 2001 From: SangamSwadiK Date: Thu, 30 Mar 2023 15:16:00 +0000 Subject: [PATCH] test --- pymc/distributions/continuous.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/pymc/distributions/continuous.py b/pymc/distributions/continuous.py index 8688cb04c2..ef341c82e2 100644 --- a/pymc/distributions/continuous.py +++ b/pymc/distributions/continuous.py @@ -814,7 +814,13 @@ class HalfNormal(PositiveContinuous): rv_op = halfnormal @classmethod - def dist(cls, sigma=None, tau=None, *args, **kwargs): + def dist( + cls, + sigma: Optional[DIST_PARAMETER_TYPES] = None, + tau: Optional[DIST_PARAMETER_TYPES] = None, + *args, + **kwargs, + ): tau, sigma = get_tau_sigma(tau=tau, sigma=sigma) return super().dist([0.0, sigma], **kwargs) @@ -946,7 +952,14 @@ class Wald(PositiveContinuous): rv_op = wald @classmethod - def dist(cls, mu=None, lam=None, phi=None, alpha=0.0, **kwargs): + def dist( + cls, + mu: Optional[DIST_PARAMETER_TYPES] = None, + lam: Optional[DIST_PARAMETER_TYPES] = None, + phi: Optional[DIST_PARAMETER_TYPES] = None, + alpha=0.0, + **kwargs, + ): mu, lam, phi = cls.get_mu_lam_phi(mu, lam, phi) alpha = pt.as_tensor_variable(floatX(alpha)) mu = pt.as_tensor_variable(floatX(mu))