Skip to content

Commit

Permalink
Add docstring2
Browse files Browse the repository at this point in the history
  • Loading branch information
julianlheureux committed Aug 28, 2024
1 parent 025ae0e commit 9d547de
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions bambi/backend/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@


def horseshoe(name, tau_nu=3, lam_nu=1, dims=None):
"""Simulate a beta coefficient value with a horseshoe prior.
"""Simulate a beta coefficient value with a horseshoe prior.
This is an internal function which is not supposed to be used by users.
This will be used only when a horseshoe prior is called for beta coefficients.
Expand All @@ -25,7 +25,7 @@ def horseshoe(name, tau_nu=3, lam_nu=1, dims=None):
Returns
------
np.ndarray
Array with the beta coefficient simulated.
Array with the beta coefficient simulated.
"""
tau = pm.HalfStudentT(f"{name}_tau", nu=tau_nu)
lam = pm.HalfStudentT(f"{name}_lam", nu=lam_nu, dims=dims)
Expand Down

0 comments on commit 9d547de

Please sign in to comment.