Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Even newer dev #106

Merged
merged 8 commits into from
Mar 17, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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
Loading