Skip to content
Snippets Groups Projects
Commit bb390ed7 authored by Martin Reinecke's avatar Martin Reinecke
Browse files

tweaks

parent 7b463c37
No related branches found
No related tags found
No related merge requests found
......@@ -23,12 +23,13 @@ class WienerFilterEnergy(Energy):
The prior signal covariance in harmonic space.
"""
def __init__(self, position, d, R, N, S):
def __init__(self, position, d, R, N, S, inverter):
super(WienerFilterEnergy, self).__init__(position=position)
self.d = d
self.R = R
self.N = N
self.S = S
self._inverter = inverter
def at(self, position):
return self.__class__(position=position, d=self.d, R=self.R, N=self.N,
......@@ -47,7 +48,8 @@ class WienerFilterEnergy(Energy):
@property
@memo
def curvature(self):
return WienerFilterCurvature(R=self.R, N=self.N, S=self.S)
return WienerFilterCurvature(R=self.R, N=self.N, S=self.S,
inverter=self._inverter)
@property
@memo
......
......@@ -78,16 +78,6 @@ def get_weight_configs():
]
def get_hermitian_configs():
np.random.seed(42)
h_0_res_real = np.random.rand(32, 16, 6).astype(np.complex128)
h_0_res_imag = np.random.rand(32, 16, 6).astype(np.complex128)
h_0_x = h_0_res_real + h_0_res_imag * 1j
return [
[h_0_x, h_0_res_real, h_0_res_imag]
]
class LMSpaceInterfaceTests(unittest.TestCase):
@expand([['lmax', int],
['mmax', int],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment