Source code for autocti.dataset_1d.fit
import autoarray as aa
from autocti.dataset_1d.dataset_1d.dataset_1d import Dataset1D
[docs]class FitDataset1D(aa.FitDataset):
def __init__(self, dataset: Dataset1D, post_cti_data):
"""
Fit a 1D CTI dataset with model cti data.
Parameters
----------
dataset
The charge injection image that is fitted.
post_cti_data
The `pre_cti_data` with cti added to it via the clocker and a CTI model.
"""
super().__init__(dataset=dataset, use_mask_in_fit=True)
self.post_cti_data = post_cti_data
@property
def dataset_1d(self) -> Dataset1D:
return self.dataset
@property
def mask(self) -> aa.Mask1D:
return self.dataset.mask
@property
def model_data(self) -> aa.Array1D:
return self.post_cti_data
@property
def pre_cti_data(self) -> aa.Array1D:
return self.dataset.pre_cti_data