Commit 95ec8d23 authored by Philipp Arras's avatar Philipp Arras
Browse files

First fixes

parent fb61bf9e
Pipeline #22070 passed with stage
in 4 minutes and 44 seconds
......@@ -4,3 +4,9 @@ from .log_normal_wiener_filter_energy import LogNormalWienerFilterEnergy
from .log_normal_wiener_filter_curvature import LogNormalWienerFilterCurvature
from .wiener_filter_energy import WienerFilterEnergy
from .wiener_filter_curvature import WienerFilterCurvature
from .noise_curvature import NoiseCurvature
from .noise_energy import NoiseEnergy
from .nonlinear_power_curvature import NonlinearPowerCurvature
from .nonlinear_power_energy import NonlinearPowerEnergy
from .nonlinear_signal_curvature import NonlinearSignalCurvature
from .nonlinear_signal_energy import NonlinearWienerFilterEnergy
from ...operators.endomorphic_operator import EndomorphicOperator
from ...operators.invertible_operator_mixin import InvertibleOperatorMixin
from ..operators.endomorphic_operator import EndomorphicOperator
class NoiseCurvature(InvertibleOperatorMixin, EndomorphicOperator):
class NoiseCurvature(EndomorphicOperator):
pass
from ... import Field, exp
from ...energies.energy import Energy
from ...memoization import memo
from ...operators.diagonal_operator import DiagonalOperator
from ...sugar import generate_posterior_sample
from .. import Field, exp
from ..operators.diagonal_operator import DiagonalOperator
from ..sugar import generate_posterior_sample
from ..minimization.energy import Energy
from ..utilities import memo
class NoiseEnergy(Energy):
......
from ... import Field
from ...operators.endomorphic_operator import EndomorphicOperator
from ...operators.invertible_operator_mixin import InvertibleOperatorMixin
from .. import Field
from ..operators.endomorphic_operator import EndomorphicOperator
from .response_operators import LinearizedPowerResponse
class NonlinearPowerCurvature(InvertibleOperatorMixin, EndomorphicOperator):
class NonlinearPowerCurvature(EndomorphicOperator):
def __init__(self, position, FFT, Instrument, nonlinearity,
Projection, N, T, sample_list, inverter=None):
......
from ... import Field, exp
from ...energies.energy import Energy
from ...memoization import memo
from ...operators.smoothness_operator import SmoothnessOperator
from ...sugar import generate_posterior_sample
from .. import Field, exp
from ..utilities import memo
from ..operators.smoothness_operator import SmoothnessOperator
from ..sugar import generate_posterior_sample
from .nonlinear_power_curvature import NonlinearPowerCurvature
from .response_operators import LinearizedPowerResponse
from ..minimization.energy import Energy
from ..operators.inversion_enabler import InversionEnabler
class NonlinearPowerEnergy(Energy):
......@@ -111,4 +112,4 @@ class NonlinearPowerEnergy(Energy):
def curvature(self):
curvature = NonlinearPowerCurvature(self.position, self.FFT, self.Instrument, self.nonlinearity,
self.Projection, self.N, self.T, self.sample_list, inverter=self.inverter)
return curvature
return InversionEnabler(curvature, inverter=self.inverter)
from ...operators.endomorphic_operator import EndomorphicOperator
from ...operators.invertible_operator_mixin import InvertibleOperatorMixin
from ..operators.endomorphic_operator import EndomorphicOperator
class NonlinearSignalCurvature(InvertibleOperatorMixin, EndomorphicOperator):
class NonlinearSignalCurvature(EndomorphicOperator):
def __init__(self, R, N, S, inverter=None):
self.R = R
self.N = N
......
from ...energies.energy import Energy
from ...memoization import memo
from .nonlinear_signal_curvature import NonlinearSignalCurvature
from .. import Field, exp
from ..utilities import memo
from ..sugar import generate_posterior_sample
from ..minimization.energy import Energy
from ..operators.inversion_enabler import InversionEnabler
from .response_operators import LinearizedSignalResponse
......@@ -68,4 +71,4 @@ class NonlinearWienerFilterEnergy(Energy):
curvature = NonlinearSignalCurvature(R=self.LinearizedResponse,
N=self.N,
S=self.S, inverter=self.inverter)
return curvature
return InversionEnabler(curvature, inverter=self.inverter)
from ... import exp
from ...operators.linear_operator import LinearOperator
from .. import exp
from ..operators.linear_operator import LinearOperator
class AdjointFFTResponse(LinearOperator):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment