Commit ece7b47f authored by Philipp Arras's avatar Philipp Arras

More renamings

parent 1bd43e72
Pipeline #31070 passed with stages
in 1 minute and 24 seconds
from .constant import ConstantModel
from .constant import Constant
from .local_nonlinearity import LocalModel
from .model import LinearModel, Model
from .position import PositionModel
from .variable import Variable
__all__ = ['Model', 'ConstantModel', 'LocalModel', 'PositionModel',
__all__ = ['Model', 'Constant', 'LocalModel', 'Variable',
'LinearModel']
......@@ -2,9 +2,9 @@ from ..operators import MultiSkyGradientOperator
from .model import Model
class ConstantModel(Model):
class Constant(Model):
def __init__(self, position, constant):
super(ConstantModel, self).__init__(position)
super(Constant, self).__init__(position)
self._constant = constant
self._value = self._constant
......
......@@ -3,12 +3,12 @@ import nifty4 as ift
from .model import Model
class PositionModel(Model):
class Variable(Model):
"""
Returns the MultiField.
"""
def __init__(self, position):
super(PositionModel, self).__init__(position)
super(Variable, self).__init__(position)
self._value = position
self._gradient = ift.ScalingOperator(1., position.domain)
......
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