Skip to content

Commit

Permalink
Merge pull request #106 from nikhil-sarin/even_newer_dev
Browse files Browse the repository at this point in the history
Even newer dev
  • Loading branch information
MoritzThomasHuebner committed Mar 17, 2022
2 parents 5bb19d3 + 07fcaec commit ef94f6f
Show file tree
Hide file tree
Showing 27 changed files with 161 additions and 63 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/python-app.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,8 @@ jobs:
python-version: 3.9
- name: Install dependencies
run: |
sudo apt install texlive-latex-extra
sudo apt-get install dvipng texlive-latex-extra texlive-fonts-recommended cm-super
python -m pip install --upgrade pip
pip install flake8 pytest
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
Expand Down
69 changes: 52 additions & 17 deletions redback/plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,16 +92,18 @@ def _get_times(self, axes: matplotlib.axes.Axes) -> np.ndarray:

@property
def xlim_low(self):
xlim_low = 0.5 * self.transient.x[0]
if xlim_low == 0:
xlim_low += 1e-3
return xlim_low
default = 0.5 * self.transient.x[0]
if default == 0:
default += 1e-3
return self.kwargs.get("xlim_low", default)

@property
def xlim_high(self):
if self.x_err is None:
return 2 * self.transient.x[-1]
return 2 * (self.transient.x[-1] + self.x_err[1][-1])
default = 2 * self.transient.x[-1]
else:
default = 2 * (self.transient.x[-1] + self.x_err[1][-1])
return self.kwargs.get("xlim_high", default)

@property
def ylim_low(self):
Expand Down Expand Up @@ -277,12 +279,32 @@ def colors(self):

@property
def xlabel(self):
return self.kwargs.get("xlabel", self.transient.xlabel)
if self.transient.use_phase_model:
default = f"Time since {self.reference_mjd_date} MJD [days]"
else:
default = self.transient.xlabel
return self.kwargs.get("xlabel", default)

@property
def ylabel(self):
return self.kwargs.get("ylabel", self.transient.ylabel)

@property
def xlim_low(self):
if self.transient.use_phase_model:
default = (self.transient.x[0] - self.reference_mjd_date) * 0.9
else:
default = 0.5 * self.transient.x[0]
return self.kwargs.get("xlim_low", default)

@property
def xlim_high(self):
if self.transient.use_phase_model:
default = (self.transient.x[-1] - self.reference_mjd_date) * 1.1
else:
default = 1.2 * self.transient.x[-1]
return self.kwargs.get("xlim_high", default)

def _get_x_err(self, indices):
return self.transient.x_err[indices] if self.transient.x_err is not None else self.transient.x_err

Expand All @@ -300,6 +322,11 @@ def _set_y_axis_multiband_data(self, axis, indices):

ncols = KwargsAccessorWithDefault("ncols", 2)

def _set_xaxis(self, axes):
if self.transient.use_phase_model:
axes.set_xscale("log")
axes.set_xlim(self.xlim_low, self.xlim_high)

@property
def nrows(self):
default = int(np.ceil(len(self.filters) / 2))
Expand All @@ -318,6 +345,12 @@ def figsize(self):
default = (4 + 4 * self.ncols, 2 + 2 * self.nrows)
return self._get_kwarg_with_default("figsize", default=default)

@property
def reference_mjd_date(self):
if self.transient.use_phase_model:
return self.kwargs.get("reference_mjd_date", int(self.transient.x[0]))
return 0

def plot_data(
self, axes: matplotlib.axes.Axes = None, save: bool = True, show: bool = True) -> None:
"""
Expand Down Expand Up @@ -349,11 +382,12 @@ def plot_data(
if isinstance(label, float):
label = f"{label:.2e}"
ax.errorbar(
self.transient.x[indices], self.transient.y[indices], xerr=self._get_x_err(indices),
yerr=self.transient.y_err[indices], fmt=self.errorbar_fmt, ms=self.ms, color=color,
self.transient.x[indices] - self.reference_mjd_date, self.transient.y[indices],
xerr=self._get_x_err(indices), yerr=self.transient.y_err[indices],
fmt=self.errorbar_fmt, ms=self.ms, color=color,
elinewidth=self.elinewidth, capsize=self.capsize, label=label)

ax.set_xlim(0.5 * self.transient.x[0], 1.2 * self.transient.x[-1])
self._set_xaxis(axes=ax)
self._set_y_axis_data(ax)

ax.set_xlabel(self.xlabel)
Expand Down Expand Up @@ -394,11 +428,11 @@ def plot_lightcurve(
frequency = redback.utils.bands_to_frequency([band])
self.model_kwargs["frequency"] = np.ones(len(times)) * frequency
ys = self.model(times, **self.max_like_params, **self.model_kwargs)
axes.plot(times, ys, color=color, alpha=0.65, lw=2)
axes.plot(times - self.reference_mjd_date, ys, color=color, alpha=0.65, lw=2)

for params in random_params:
ys = self.model(times, **params, **self.model_kwargs)
axes.plot(times, ys, color='red', alpha=0.05, lw=2, zorder=-1)
axes.plot(times - self.reference_mjd_date, ys, color='red', alpha=0.05, lw=2, zorder=-1)

self._save_and_show(filepath=self.lightcurve_plot_filepath, save=save, show=show)
return axes
Expand Down Expand Up @@ -455,11 +489,12 @@ def plot_multiband(

label = self._get_multiband_plot_label(band, freq)
axes[i].errorbar(
self.transient.x[indices], self.transient.y[indices], xerr=x_err, yerr=self.transient.y_err[indices],
fmt=self.errorbar_fmt, ms=self.ms, color=color, elinewidth=self.elinewidth, capsize=self.capsize,
self.transient.x[indices] - self.reference_mjd_date, self.transient.y[indices], xerr=x_err,
yerr=self.transient.y_err[indices], fmt=self.errorbar_fmt, ms=self.ms, color=color,
elinewidth=self.elinewidth, capsize=self.capsize,
label=label)

axes[i].set_xlim(0.5 * self.transient.x[indices][0], 1.2 * self.transient.x[indices][-1])
self._set_xaxis(axes[i])
self._set_y_axis_multiband_data(axes[i], indices)
axes[i].legend(ncol=2)
axes[i].tick_params(axis='both', which='major', pad=8)
Expand Down Expand Up @@ -525,10 +560,10 @@ def plot_multiband_lightcurve(

for i in range(len(ys)):
axes[i].plot(
times, ys[i], color=self.max_likelihood_color, alpha=self.max_likelihood_alpha, lw=self.linewidth)
times - self.reference_mjd_date, ys[i], color=self.max_likelihood_color, alpha=self.max_likelihood_alpha, lw=self.linewidth)
for random_ys in random_ys_list:
axes[i].plot(
times, random_ys[i], color=self.random_sample_color,
times - self.reference_mjd_date, random_ys[i], color=self.random_sample_color,
alpha=self.random_sample_alpha, lw=self.linewidth, zorder=self.zorder)

self._save_and_show(filepath=self.multiband_lightcurve_plot_filepath, save=save, show=show)
Expand Down
2 changes: 1 addition & 1 deletion redback/priors/basic_magnetar_powered.prior
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
redshift = Uniform(1e-6,3,name='redshift', latex_label = r'$z$')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0}$ [ms]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}$~[10$^{14}$G]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}$ [10$^{14}$G]')
mass_ns = Uniform(1.1, 2.2, 'mass_ns', latex_label = r'$M_{\mathrm{NS}} [M_{\odot}]$')
theta_pb = Uniform(0, 3.14/2, 'theta_pb', latex_label = r'$\\theta_{P-B}$')
mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej}} [M_{\odot}]$')
Expand Down
8 changes: 4 additions & 4 deletions redback/priors/basic_magnetar_powered_bolometric.prior
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0}$ [ms]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}$~[10$^{14}$G]')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0}\,$ [ms]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}\,$ [10$^{14}$G]')
mass_ns = Uniform(1.1, 2.2, 'mass_ns', latex_label = r'$M_{\mathrm{NS}} [M_{\odot}]$')
theta_pb = Uniform(0, 3.14/2, 'theta_pb', latex_label = r'$\\theta_{P-B}$')
mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej}} [M_{\odot}]$')
vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}} [km/s]$')
mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej}}\, [M_{\odot}]$')
vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}}\,$ [km/s]')
kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\kappa$')
kappa_gamma = Uniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\kappa_{\gamma}$')
2 changes: 1 addition & 1 deletion redback/priors/collapsing_magnetar.prior
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label= r'$n$')
tcol = LogUniform(10, 1e8, 'tcol', latex_label= r'$t_\mathrm{col}$')
2 changes: 1 addition & 1 deletion redback/priors/collapsing_radiative_losses.prior
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label= r'$n$')
tcol = LogUniform(10, 1e8, 'tcol', latex_label= r'$t_\mathrm{col}$')
kappa = LogUniform(1e-2, 10, 'kappa', latex_label= r'$\kappa$')
Expand Down
8 changes: 4 additions & 4 deletions redback/priors/five_component_fireball.prior
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta t_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta_{ } t_{1}$')
alpha_2 = Uniform(-10, 0, 'alpha_2', latex_label = r'$\\alpha_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta t_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta_{ } t_{2}$')
alpha_3 = Uniform(-10, 0, 'alpha_3', latex_label = r'$\\alpha_{3}$')
delta_time_three = LogUniform(1e-10, 1e6, 'delta_time_three', latex_label = r'$\Delta t_{3}$')
delta_time_three = LogUniform(1e-10, 1e6, 'delta_time_three', latex_label = r'$\Delta_{ } t_{3}$')
alpha_4 = Uniform(-10, 0, 'alpha_4', latex_label = r'$\\alpha_{4}$')
delta_time_four = LogUniform(1e-10, 1e6, 'delta_time_four', latex_label = r'$\Delta t_{4}$')
delta_time_four = LogUniform(1e-10, 1e6, 'delta_time_four', latex_label = r'$\Delta_{ } t_{4}$')
alpha_5 = Uniform(-10, 0, 'alpha_5', latex_label = r'$\\alpha_{5}$')
4 changes: 2 additions & 2 deletions redback/priors/fixed_braking_index.prior
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
amplitude_one = LogUniform(1e-1, 1e5, 'amplitude_one', latex_label = r'$A_{1}$')
time_exponent_one = Uniform(-10, 0, 'time_exponent_one', latex_label = r'$\\alpha_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta t_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta_{ } t_{1}$')
time_exponent_two = -2.0
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta t_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta_{ } t_{2}$')
6 changes: 3 additions & 3 deletions redback/priors/four_component_fireball.prior
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta t_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta_{ } t_{1}$')
alpha_2 = Uniform(-10, 0, 'alpha_2', latex_label = r'$\\alpha_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta t_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta_{ } t_{2}$')
alpha_3 = Uniform(-10, 0, 'alpha_3', latex_label = r'$\\alpha_{3}$')
delta_time_three = LogUniform(1e-10, 1e6, 'delta_time_three', latex_label = r'$\Delta t_{3}$')
delta_time_three = LogUniform(1e-10, 1e6, 'delta_time_three', latex_label = r'$\Delta_{ } t_{3}$')
alpha_4 = Uniform(-10, 0, 'alpha_4', latex_label = r'$\\alpha_{4}$')
2 changes: 1 addition & 1 deletion redback/priors/full_magnetar.prior
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label = r'$n$')
4 changes: 2 additions & 2 deletions redback/priors/general_magnetar.prior
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-5, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta t_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta_{ } t_{1}$')
alpha_2 = Uniform(-10, 0, 'alpha_2', latex_label = r'$\\alpha_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta t_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta_{ } t_{2}$')
4 changes: 2 additions & 2 deletions redback/priors/gw_magnetar.prior
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
fgw0 = LogUniform(100, 3000, 'fgw0', latex_label = r'$f_{gw,0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
fgw0 = LogUniform(100, 3000, 'fgw0', latex_label = r'$f_{gw, 0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label = r'$n$')
logII = Uniform(44,46,'logII', latex_label = r'$\log_{10}I$')
4 changes: 2 additions & 2 deletions redback/priors/magnetar_nickel.prior
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
redshift = Uniform(1e-6,3,name='redshift', latex_label = r'$z$')
f_nickel = LogUniform(1e-3,1,name='f_nickel', latex_label = r'$f_{\mathrm{Ni}}$')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0]$ [ms]')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0}$ [ms]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}$~[10$^{14}$G]')
mass_ns = Uniform(1.1, 2.2, 'mass_ns', latex_label = r'$M_{\mathrm{NS}} [M_{\odot}]$')
theta_pb = Uniform(0, 3.14/2, 'theta_pb', latex_label = r'$\\theta_{P-B}$')
mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej}} [M_{\odot}]$')
vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}} [km/s]$')
kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\kappa$')
kappa_gamma = Uniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\kappa_{\gamma}$')
temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}$ [k]}')
temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}$ [k]')
2 changes: 1 addition & 1 deletion redback/priors/magnetar_only.prior
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label = r'$n$')
2 changes: 1 addition & 1 deletion redback/priors/mergernova.prior
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,6 @@ beta = Uniform(0.1, 0.7, 'beta', latex_label = r'$v_{\mathrm{ej}} [c]$')
kappa = Uniform(1, 30, 'kappa', latex_label = r'$\kappa$')
n_ism = LogUniform(1e-4, 1, 'n_ism', latex_label = r'$m_{\mathrm{ism}}$')
l0 = LogUniform(1e40, 1e50, 'l0', latex_label = r'$l_0$')
tau_sd = LogUniform(10, 1e5, 'tau_sd', latex_label = '$\tau_{\mathrm{sd}}$')
tau_sd = LogUniform(10, 1e5, 'tau_sd', latex_label = '$\\tau_{\mathrm{sd}}$')
nn = Uniform(2, 7, 'nn', latex_label='$n$')
thermalisation_efficiency = Uniform(0.1, 1, 'thermalisation_efficiency', latex_label='$\epsilon_{\mathrm{th}}$')
2 changes: 1 addition & 1 deletion redback/priors/piecewise_radiative_losses.prior
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label= r'$n$')
kappa = LogUniform(1e-2, 10, 'kappa', latex_label= r'$\kappa$')
t0 = Uniform(1, 1000, 't0', latex_label= r'$T_{0}$')
2 changes: 1 addition & 1 deletion redback/priors/radiative_losses.prior
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
nn = Uniform(1.1, 7, 'nn', latex_label= r'$n$')
kappa = LogUniform(1e-2, 10, 'kappa', latex_label= r'$\kappa$')
t0 = Uniform(1, 1000, 't0', latex_label= r'$T_{0}$')
Expand Down
2 changes: 1 addition & 1 deletion redback/priors/radiative_losses_mdr.prior
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
l0 = LogUniform(1e-20, 1e5, 'l0', latex_label = r'$L_{0}$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\tau$')
tau = LogUniform(1e2, 1e6, 'tau', latex_label = r'$\\tau$')
kappa = LogUniform(1e-2, 10, 'kappa', latex_label= r'$\kappa$')
t0 = Uniform(1, 1000, 't0', latex_label= r'$T_{0}$')
log_e0 = Uniform(-10, 2, 'log_e0', latex_label = r'$\log_{10} E_{0}$')
10 changes: 5 additions & 5 deletions redback/priors/six_component_fireball.prior
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
a_1 = LogUniform(1e-20, 1e20, 'a_1', latex_label = r'$A_{1}$')
alpha_1 = Uniform(-10, -0.5, 'alpha_1', latex_label = r'$\\alpha_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta t_{1}$')
delta_time_one = LogUniform(1e-10, 1e6, 'delta_time_one', latex_label = r'$\Delta_{ } t_{1}$')
alpha_2 = Uniform(-10, 0, 'alpha_2', latex_label = r'$\\alpha_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta t_{2}$')
delta_time_two = LogUniform(10, 1e6, 'delta_time_two', latex_label = r'$\Delta_{ } t_{2}$')
alpha_3 = Uniform(-10, 0, 'alpha_3', latex_label = r'$\\alpha_{3}$')
delta_time_three = LogUniform(1e-10, 1e6, 'delta_time_three', latex_label = r'$\Delta t_{3}$')
delta_time_three = LogUniform(1e-10, 1e6, 'delta_time_three', latex_label = r'$\Delta_{ } t_{3}$')
alpha_4 = Uniform(-10, 0, 'alpha_4', latex_label = r'$\\alpha_{4}$')
delta_time_four = LogUniform(1e-10, 1e6, 'delta_time_four', latex_label = r'$\Delta t_{4}$')
delta_time_four = LogUniform(1e-10, 1e6, 'delta_time_four', latex_label = r'$\Delta_{ } t_{4}$')
alpha_5 = Uniform(-10, 0, 'alpha_5', latex_label = r'$\\alpha_{5}$')
delta_time_five = LogUniform(1e-10, 1e6, 'delta_time_five', latex_label = r'$\Delta t_{5}$')
delta_time_five = LogUniform(1e-10, 1e6, 'delta_time_five', latex_label = r'$\Delta_{ } t_{5}$')
alpha_six = Uniform(-10, 0, 'alpha_six', latex_label = r'$\\alpha_{6}$')
4 changes: 2 additions & 2 deletions redback/priors/slsn.prior
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
redshift = Uniform(1e-6,3,name='redshift', latex_label = r'$z$')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0]$ [ms]')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0}$ [ms]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}$~[10$^{14}$G]')
mass_ns = Uniform(1.1, 2.2, 'mass_ns', latex_label = r'$M_{\mathrm{NS}} [M_{\odot}]$')
theta_pb = Uniform(0, 3.14/2, 'theta_pb', latex_label = r'$\\theta_{P-B}$')
mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej}} [M_{\odot}]$')
vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}} [km/s]$')
kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\kappa$')
kappa_gamma = Uniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\kappa_{\gamma}$')
temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}$ [k]}')
temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}$ [k]')
e_rot_constraint = Constraint(name='e_rot_constraint', minimum=10, maximum=1e10)
t_nebula_min = Constraint(name='t_nebula_min', minimum=0.1,maximum=500)
2 changes: 1 addition & 1 deletion redback/priors/slsn_bolometric.prior
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0]$ [ms]')
p0 = Uniform(1, 10, 'p0', latex_label = r'$P_{0}$ [ms]')
bp = LogUniform(0.1,10,'bp',latex_label = r'$B_{p}$~[10$^{14}$G]')
mass_ns = Uniform(1.1, 2.2, 'mass_ns', latex_label = r'$M_{\mathrm{NS}} [M_{\odot}]$')
theta_pb = Uniform(0, 3.14/2, 'theta_pb', latex_label = r'$\\theta_{P-B}$')
Expand Down

0 comments on commit ef94f6f

Please sign in to comment.