Skip to content

Commit

Permalink
extend compatibility to astropy4
Browse files Browse the repository at this point in the history
  • Loading branch information
ankitkhushwaha committed Feb 25, 2025
1 parent aa8da42 commit 8bb2e6f
Showing 1 changed file with 22 additions and 22 deletions.
44 changes: 22 additions & 22 deletions stingray/simulator/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,18 +95,18 @@ def bounding_box(self, factor=25):

return (x0 - dx, x0 + dx)

@property
def input_units(self):
if self.x_0.input_unit is None:
return None
return {self.inputs[0]: self.x_0.input_unit}
# @property
# def input_units(self):
# if self.x_0.input_unit is None:
# return None
# return {self.inputs[0]: self.x_0.input_unit}

def _parameter_units_for_data_units(self, inputs_unit, outputs_unit):
return {
"x_0": inputs_unit[self.inputs[0]],
"fwhm": inputs_unit[self.inputs[0]],
"value": outputs_unit[self.outputs[0]],
}
# def _parameter_units_for_data_units(self, inputs_unit, outputs_unit):
# return {
# "x_0": inputs_unit[self.inputs[0]],
# "fwhm": inputs_unit[self.inputs[0]],
# "value": outputs_unit[self.outputs[0]],
# }


class SmoothBrokenPowerLaw(Fittable1DModel):
Expand Down Expand Up @@ -190,17 +190,17 @@ def fit_deriv(x, norm, gamma_low, gamma_high, break_freq):
)
return [del_func_x, del_func_norm, del_func_g_low, del_func_g_high, del_func_b_freq]

@property
def input_units(self):
if self.break_freq.input_unit is None:
return None
return {self.inputs[0]: self.break_freq.input_unit}

def _parameter_units_for_data_units(self, inputs_unit, outputs_unit):
return {
"break_freq": inputs_unit[self.inputs[0]],
"norm_": outputs_unit[self.outputs[0]],
}
# @property
# def input_units(self):
# if self.break_freq.input_unit is None:
# return None
# return {self.inputs[0]: self.break_freq.input_unit}

# def _parameter_units_for_data_units(self, inputs_unit, outputs_unit):
# return {
# "break_freq": inputs_unit[self.inputs[0]],
# "norm_": outputs_unit[self.outputs[0]],
# }


def generalized_lorentzian(x, p):
Expand Down

0 comments on commit 8bb2e6f

Please sign in to comment.