diff --git a/tupak/gw/prior.py b/tupak/gw/prior.py index d45f2a7eedc314242aa0618ecdabc60a416a0309..eaa0eb561617253f448cacf9720932180b233299 100644 --- a/tupak/gw/prior.py +++ b/tupak/gw/prior.py @@ -8,7 +8,8 @@ from scipy.interpolate import UnivariateSpline class UniformComovingVolume(FromFile): - def __init__(self, minimum=None, maximum=None, name='luminosity distance', latex_label='$d_L$'): + def __init__(self, minimum=None, maximum=None, + name='luminosity distance', latex_label='$d_L$', unit='Mpc'): """ Parameters @@ -24,7 +25,7 @@ class UniformComovingVolume(FromFile): """ file_name = os.path.join(os.path.dirname(__file__), 'prior_files', 'comoving.txt') FromFile.__init__(self, file_name=file_name, minimum=minimum, maximum=maximum, name=name, - latex_label=latex_label, unit='Mpc') + latex_label=latex_label, unit=unit) class AlignedSpin(Interped):