Commit 18a751dc authored by Philipp Arras's avatar Philipp Arras
Browse files

Fixes

parent 6eee9ecf
Pipeline #70626 failed with stages
in 13 minutes and 16 seconds
......@@ -15,7 +15,6 @@ from .multi_domain import MultiDomain
from .field import Field
from .multi_field import MultiField
from .linearization import Linearization
from .operators.operator import Operator
from .operators.linear_operator import LinearOperator
......@@ -70,6 +69,7 @@ from .minimization.energy_adapter import EnergyAdapter
from .minimization.metric_gaussian_kl import MetricGaussianKL
from .sugar import *
from .plot import Plot
from .library.special_distributions import InverseGammaOperator
......@@ -90,6 +90,8 @@ from .utilities import memo, frozendict
from .logger import logger
from .linearization import Linearization
from .operator_spectrum import operator_spectrum
# We deliberately don't set __all__ here, because we don't want people to do a
......
......@@ -133,7 +133,7 @@ class LightConeOperator(Operator):
def apply(self, x):
lin = isinstance(x, Linearization)
a, derivs = _cone_arrays(x.val, self.target, self._sigx, lin)
a, derivs = _cone_arrays(x.val.val if lin else x.val, self.target, self._sigx, lin)
res = Field(self.target, a)
if not lin:
return res
......
Supports Markdown
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