__init__.py 4.15 KB
Newer Older
1
from .version import __version__
Martin Reinecke's avatar
Martin Reinecke committed
2
3

from . import dobj
Martin Reinecke's avatar
Martin Reinecke committed
4
5
6
7
8
9
10
11
12
13
14
15

from .domains.domain import Domain
from .domains.structured_domain import StructuredDomain
from .domains.unstructured_domain import UnstructuredDomain
from .domains.rg_space import RGSpace
from .domains.lm_space import LMSpace
from .domains.gl_space import GLSpace
from .domains.hp_space import HPSpace
from .domains.power_space import PowerSpace
from .domains.dof_space import DOFSpace
from .domains.log_rg_space import LogRGSpace

16
from .domain_tuple import DomainTuple
Martin Reinecke's avatar
Martin Reinecke committed
17
from .multi_domain import MultiDomain
18
from .field import Field
Martin Reinecke's avatar
Martin Reinecke committed
19
from .multi_field import MultiField
Philipp Arras's avatar
Fixups    
Philipp Arras committed
20

Martin Reinecke's avatar
Martin Reinecke committed
21
from .operators.operator import Operator
Martin Reinecke's avatar
Martin Reinecke committed
22
from .operators.diagonal_operator import DiagonalOperator
Martin Reinecke's avatar
cleanup    
Martin Reinecke committed
23
from .operators.distributors import DOFDistributor, PowerDistributor
Philipp Arras's avatar
Philipp Arras committed
24
from .operators.domain_tuple_field_inserter import DomainTupleFieldInserter
25
from .operators.contraction_operator import ContractionOperator
26
from .operators.linear_interpolation import LinearInterpolator
Martin Reinecke's avatar
Martin Reinecke committed
27
28
from .operators.endomorphic_operator import EndomorphicOperator
from .operators.exp_transform import ExpTransform
Martin Reinecke's avatar
Martin Reinecke committed
29
30
31
from .operators.harmonic_operators import (
    FFTOperator, HartleyOperator, SHTOperator, HarmonicTransformOperator,
    HarmonicSmoothingOperator)
Martin Reinecke's avatar
Martin Reinecke committed
32
from .operators.field_zero_padder import FieldZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
33
34
35
from .operators.inversion_enabler import InversionEnabler
from .operators.linear_operator import LinearOperator
from .operators.mask_operator import MaskOperator
Philipp Arras's avatar
Philipp Arras committed
36
from .operators.offset_operator import OffsetOperator
Martin Reinecke's avatar
Martin Reinecke committed
37
from .operators.qht_operator import QHTOperator
38
from .operators.regridding_operator import RegriddingOperator
Martin Reinecke's avatar
Martin Reinecke committed
39
40
41
42
43
from .operators.sampling_enabler import SamplingEnabler
from .operators.sandwich_operator import SandwichOperator
from .operators.scaling_operator import ScalingOperator
from .operators.slope_operator import SlopeOperator
from .operators.symmetrizing_operator import SymmetrizingOperator
Martin Reinecke's avatar
Martin Reinecke committed
44
from .operators.block_diagonal_operator import BlockDiagonalOperator
45
from .operators.outer_product_operator import OuterProduct
Martin Reinecke's avatar
Martin Reinecke committed
46
from .operators.simple_linear_operators import (
47
    VdotOperator, ConjugationOperator, Realizer,
48
    FieldAdapter, ducktape, GeometryRemover, NullOperator)
Philipp Arras's avatar
Philipp Arras committed
49
from .operators.value_inserter import ValueInserter
Martin Reinecke's avatar
Martin Reinecke committed
50
from .operators.energy_operators import (
Martin Reinecke's avatar
merge    
Martin Reinecke committed
51
    EnergyOperator, GaussianEnergy, PoissonianEnergy, InverseGammaLikelihood,
Martin Reinecke's avatar
Martin Reinecke committed
52
    BernoulliEnergy, Hamiltonian, AveragedEnergy)
Martin Reinecke's avatar
Martin Reinecke committed
53

Martin Reinecke's avatar
Martin Reinecke committed
54
from .probing import probe_with_posterior_samples, probe_diagonal, \
55
    StatCalculator
Martin Reinecke's avatar
Martin Reinecke committed
56
57

from .minimization.line_search import LineSearch
Martin Reinecke's avatar
Martin Reinecke committed
58
from .minimization.iteration_controllers import (
Martin Reinecke's avatar
Martin Reinecke committed
59
60
    IterationController, GradientNormController, DeltaEnergyController,
    GradInfNormController)
Martin Reinecke's avatar
Martin Reinecke committed
61
62
63
from .minimization.minimizer import Minimizer
from .minimization.conjugate_gradient import ConjugateGradient
from .minimization.nonlinear_cg import NonlinearCG
64
from .minimization.descent_minimizers import (
65
66
67
    DescentMinimizer, SteepestDescent, VL_BFGS, L_BFGS, RelaxedNewton,
    NewtonCG)
from .minimization.scipy_minimizer import (ScipyMinimizer, L_BFGS_B, ScipyCG)
Martin Reinecke's avatar
Martin Reinecke committed
68
69
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
Martin Reinecke's avatar
Martin Reinecke committed
70
from .minimization.energy_adapter import EnergyAdapter
71
from .minimization.kl_energy import KL_Energy
Martin Reinecke's avatar
Martin Reinecke committed
72

Martin Reinecke's avatar
Martin Reinecke committed
73
from .sugar import *
Martin Reinecke's avatar
Martin Reinecke committed
74
from .plot import Plot
Philipp Arras's avatar
Fixups    
Philipp Arras committed
75

Philipp Arras's avatar
Docs    
Philipp Arras committed
76
from .library.smooth_linear_amplitude import (SLAmplitude, CepstrumOperator)
Philipp Arras's avatar
Philipp Arras committed
77
from .library.inverse_gamma_operator import InverseGammaOperator
Martin Reinecke's avatar
Martin Reinecke committed
78
from .library.los_response import LOSResponse
Philipp Arras's avatar
Philipp Arras committed
79
80
from .library.dynamic_operator import (dynamic_operator,
                                       dynamic_lightcone_operator)
81
from .library.light_cone_operator import LightConeOperator
Martin Reinecke's avatar
merge    
Martin Reinecke committed
82

Philipp Arras's avatar
Fixup    
Philipp Arras committed
83
from .library.wiener_filter_curvature import WienerFilterCurvature
84
from .library.correlated_fields import CorrelatedField, MfCorrelatedField
Martin Reinecke's avatar
Martin Reinecke committed
85
86
from .library.adjust_variances import (make_adjust_variances,
                                       do_adjust_variances)
Martin Reinecke's avatar
Martin Reinecke committed
87

Philipp Arras's avatar
Move it    
Philipp Arras committed
88
from . import extra
89

Martin Reinecke's avatar
Martin Reinecke committed
90
from .utilities import memo, frozendict
Martin Reinecke's avatar
Martin Reinecke committed
91

Martin Reinecke's avatar
Martin Reinecke committed
92
93
from .logger import logger

Martin Reinecke's avatar
Martin Reinecke committed
94
from .linearization import Linearization
95

96
# We deliberately don't set __all__ here, because we don't want people to do a
Philipp Arras's avatar
Philipp Arras committed
97
# "from nifty5 import *"; that would swamp the global namespace.