def obs_data(self, data):
if self._grad_data_type in (float, np.floating):
data = check_float(data)
check_npndarray(data, dtype=self._grad_data_type, writeable=False)
self._obs_data = data
def obs_data(self, input_data):
if self._grad_data_type in {float, np.floating}:
input_data = check_float(input_data)
check_npndarray(
input_data,
dtype=self._grad_data_type,
writeable=False,
verbose=self.verbose,
)
self._obs_data = input_data
def __init__(self, data, psf, psf_type='fixed', convolve_method='astropy'):
self.obs_data = data
self.op = self._H_op_method
self.trans_op = self._Ht_op_method
check_float(psf)
check_npndarray(psf, writeable=False)
self._psf = psf
self._psf_type = psf_type
self._convolve_method = convolve_method
PowerMethod.__init__(self, self.trans_op_op, self.obs_data.shape)
def check_data_format(data, n_dim):
"""Check data format
This method checks that the input data has the correct number of dimensions
Parameters
----------
data : np.ndarray
Input data array
n_dim : int or list of ints
Expected number of dimensions
Raises
------
ValueError
For invalid array dimensions
"""
check_npndarray(data, dtype=float, writeable=False, verbose=False)
if data.ndim not in list(n_dim):
raise ValueError('Input data array has an invalid number of '
'dimensions.')
