diff --git a/bilby/core/prior/base.py b/bilby/core/prior/base.py index ea8607545c59126ab7e5c4d6e83b8c5ba7ba83d8..e11f7862ad7360737760eaa272cd6157f85d8ba6 100644 --- a/bilby/core/prior/base.py +++ b/bilby/core/prior/base.py @@ -214,10 +214,13 @@ class Prior(object): """ prior_name = self.__class__.__name__ + prior_module = self.__class__.__module__ instantiation_dict = self.get_instantiation_dict() - args = ', '.join(['{}={}'.format(key, repr(instantiation_dict[key])) - for key in instantiation_dict]) - return "{}({})".format(prior_name, args) + args = ', '.join([f'{key}={repr(instantiation_dict[key])}' for key in instantiation_dict]) + if "bilby.core.prior" in prior_module: + return f"{prior_name}({args})" + else: + return f"{prior_module}.{prior_name}({args})" @property def _repr_dict(self): diff --git a/test/core/prior/prior_test.py b/test/core/prior/prior_test.py index 9241945433cca240af4a4dc740b537d24aa59c83..3fd77af938a4abf8fad832a669e89f797ea87f5f 100644 --- a/test/core/prior/prior_test.py +++ b/test/core/prior/prior_test.py @@ -673,7 +673,7 @@ class TestPriorClasses(unittest.TestCase): "bilby.core.prior.MultivariateGaussianDist", ) elif isinstance(prior, bilby.gw.prior.HealPixPrior): - repr_prior_string = "bilby.gw.prior." + repr(prior) + repr_prior_string = repr(prior) repr_prior_string = repr_prior_string.replace( "HealPixMapPriorDist", "bilby.gw.prior.HealPixMapPriorDist" )