Skip to content
Snippets Groups Projects
Commit a242eed0 authored by moritz's avatar moritz
Browse files

Moritz Huebner: Minor code clean up

parent 460fcfe8
No related branches found
No related tags found
No related merge requests found
Pipeline #
from __future__ import division, print_function
import numpy as np
import peyote
from . import prior
class Parameter:
......@@ -71,37 +71,37 @@ class Parameter:
def set_default_prior(self):
if self.name == 'mass_1':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(5, 100))
self.__prior = prior.PowerLaw(alpha=0, bounds=(5, 100))
elif self.name == 'mass_2':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(5, 100))
self.__prior = prior.PowerLaw(alpha=0, bounds=(5, 100))
elif self.name == 'mchirp':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(5, 100))
self.__prior = prior.PowerLaw(alpha=0, bounds=(5, 100))
elif self.name == 'q':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 1))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 1))
elif self.name == 'a1':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 1))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 1))
elif self.name == 'a2':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 1))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 1))
elif self.name == 'tilt1':
self.__prior = peyote.prior.Sine()
self.__prior = prior.Sine()
elif self.name == 'tilt2':
self.__prior = peyote.prior.Sine()
self.__prior = prior.Sine()
elif self.name == 'phi1':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
elif self.name == 'phi2':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
elif self.name == 'luminosity_distance':
self.__prior = peyote.prior.PowerLaw(alpha=2, bounds=(1e2, 5e3))
self.__prior = prior.PowerLaw(alpha=2, bounds=(1e2, 5e3))
elif self.name == 'dec':
self.__prior = peyote.prior.Cosine()
self.__prior = prior.Cosine()
elif self.name == 'ra':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
elif self.name == 'iota':
self.__prior = peyote.prior.Sine()
self.__prior = prior.Sine()
elif self.name == 'psi':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
elif self.name == 'phase':
self.__prior = peyote.prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
self.__prior = prior.PowerLaw(alpha=0, bounds=(0, 2 * np.pi))
else:
self.__prior = None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment