diff --git a/nifty4/library/los_response.py b/nifty4/library/los_response.py index 0c25765521ce7e21b75d747af87004a2f73bc067..a28105cdc8dde91c449306536d5c33d8a7f16483 100644 --- a/nifty4/library/los_response.py +++ b/nifty4/library/los_response.py @@ -147,10 +147,14 @@ class LOSResponse(LinearOperator): sigmas_up = np.zeros(nlos, dtype=np.float32) sigmas_low = np.array(sigmas_low) sigmas_up = np.array(sigmas_up) - assert starts.shape[0] == ndim, "dimension mismatch" - assert nlos == sigmas_low.shape[0], "dimension mismatch" - assert starts.shape == ends.shape, "dimension mismatch" - assert sigmas_low.shape == sigmas_up.shape, "dimension mismatch" + if starts.shape[0] != ndim: + raise TypeError("dimension mismatch") + if nlos != sigmas_low.shape[0]: + raise TypeError("dimension mismatch") + if starts.shape != ends.shape: + raise TypeError("dimension mismatch") + if sigmas_low.shape != sigmas_up.shape: + raise TypeError("dimension mismatch") self._local_shape = dobj.local_shape(self.domain[0].shape) local_zero_point = (np.array(