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

Martin Reinecke's avatar
Martin Reinecke committed
3
4
5
6
7
8
9
10
11
12
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

13
from .domain_tuple import DomainTuple
Martin Reinecke's avatar
Martin Reinecke committed
14
from .multi_domain import MultiDomain
15
from .field import Field
Martin Reinecke's avatar
Martin Reinecke committed
16
from .multi_field import MultiField
Philipp Arras's avatar
Fixups    
Philipp Arras committed
17

Martin Reinecke's avatar
Martin Reinecke committed
18
from .operators.operator import Operator
Philipp Arras's avatar
Philipp Arras committed
19
from .operators.adder import Adder
Martin Reinecke's avatar
Martin Reinecke committed
20
from .operators.diagonal_operator import DiagonalOperator
Martin Reinecke's avatar
cleanup    
Martin Reinecke committed
21
from .operators.distributors import DOFDistributor, PowerDistributor
Philipp Arras's avatar
Philipp Arras committed
22
from .operators.domain_tuple_field_inserter import DomainTupleFieldInserter
23
from .operators.contraction_operator import ContractionOperator
24
from .operators.linear_interpolation import LinearInterpolator
Martin Reinecke's avatar
Martin Reinecke committed
25
from .operators.endomorphic_operator import EndomorphicOperator
Martin Reinecke's avatar
Martin Reinecke committed
26
27
28
from .operators.harmonic_operators import (
    FFTOperator, HartleyOperator, SHTOperator, HarmonicTransformOperator,
    HarmonicSmoothingOperator)
Martin Reinecke's avatar
Martin Reinecke committed
29
from .operators.field_zero_padder import FieldZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
30
31
32
from .operators.inversion_enabler import InversionEnabler
from .operators.linear_operator import LinearOperator
from .operators.mask_operator import MaskOperator
33
from .operators.regridding_operator import RegriddingOperator
Martin Reinecke's avatar
Martin Reinecke committed
34
35
36
from .operators.sampling_enabler import SamplingEnabler
from .operators.sandwich_operator import SandwichOperator
from .operators.scaling_operator import ScalingOperator
Martin Reinecke's avatar
Martin Reinecke committed
37
from .operators.block_diagonal_operator import BlockDiagonalOperator
38
from .operators.outer_product_operator import OuterProduct
Martin Reinecke's avatar
Martin Reinecke committed
39
from .operators.simple_linear_operators import (
40
    VdotOperator, ConjugationOperator, Realizer,
41
    FieldAdapter, ducktape, GeometryRemover, NullOperator,
Philipp Arras's avatar
Philipp Arras committed
42
    MatrixProductOperator, PartialExtractor)
Philipp Arras's avatar
Philipp Arras committed
43
from .operators.value_inserter import ValueInserter
Martin Reinecke's avatar
Martin Reinecke committed
44
from .operators.energy_operators import (
Martin Reinecke's avatar
merge    
Martin Reinecke committed
45
    EnergyOperator, GaussianEnergy, PoissonianEnergy, InverseGammaLikelihood,
46
    BernoulliEnergy, StandardHamiltonian, AveragedEnergy, QuadraticFormOperator,
47
    Squared2NormOperator, StudentTEnergy)
48
from .operators.convolution_operators import FuncConvolutionOperator
Martin Reinecke's avatar
Martin Reinecke committed
49

Martin Reinecke's avatar
Martin Reinecke committed
50
from .probing import probe_with_posterior_samples, probe_diagonal, \
51
    StatCalculator, approximation2endo
Martin Reinecke's avatar
Martin Reinecke committed
52
53

from .minimization.line_search import LineSearch
Martin Reinecke's avatar
Martin Reinecke committed
54
from .minimization.iteration_controllers import (
Martin Reinecke's avatar
Martin Reinecke committed
55
    IterationController, GradientNormController, DeltaEnergyController,
56
    GradInfNormController, AbsDeltaEnergyController)
Martin Reinecke's avatar
Martin Reinecke committed
57
58
59
from .minimization.minimizer import Minimizer
from .minimization.conjugate_gradient import ConjugateGradient
from .minimization.nonlinear_cg import NonlinearCG
60
from .minimization.descent_minimizers import (
61
62
    DescentMinimizer, SteepestDescent, VL_BFGS, L_BFGS, RelaxedNewton,
    NewtonCG)
Martin Reinecke's avatar
Martin Reinecke committed
63
from .minimization.scipy_minimizer import L_BFGS_B
Martin Reinecke's avatar
Martin Reinecke committed
64
65
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
Martin Reinecke's avatar
Martin Reinecke committed
66
from .minimization.energy_adapter import EnergyAdapter
67
from .minimization.metric_gaussian_kl import MetricGaussianKL
Martin Reinecke's avatar
Martin Reinecke committed
68

Martin Reinecke's avatar
Martin Reinecke committed
69
from .sugar import *
70
from .plot import Plot
Philipp Arras's avatar
Fixups    
Philipp Arras committed
71

Philipp Arras's avatar
Philipp Arras committed
72
from .library.inverse_gamma_operator import InverseGammaOperator
Martin Reinecke's avatar
Martin Reinecke committed
73
from .library.los_response import LOSResponse
Philipp Arras's avatar
Philipp Arras committed
74
75
from .library.dynamic_operator import (dynamic_operator,
                                       dynamic_lightcone_operator)
76
from .library.light_cone_operator import LightConeOperator
Martin Reinecke's avatar
merge    
Martin Reinecke committed
77

Philipp Arras's avatar
Fixup    
Philipp Arras committed
78
from .library.wiener_filter_curvature import WienerFilterCurvature
79
from .library.adjust_variances import (make_adjust_variances_hamiltonian,
Martin Reinecke's avatar
Martin Reinecke committed
80
                                       do_adjust_variances)
Philipp Arras's avatar
Philipp Arras committed
81
from .library.gridder import GridderMaker
Philipp Arras's avatar
Philipp Arras committed
82
from .library.correlated_fields import CorrelatedFieldMaker
Martin Reinecke's avatar
Martin Reinecke committed
83

Philipp Arras's avatar
Move it    
Philipp Arras committed
84
from . import extra
85

Martin Reinecke's avatar
Martin Reinecke committed
86
from .utilities import memo, frozendict
Martin Reinecke's avatar
Martin Reinecke committed
87

Martin Reinecke's avatar
Martin Reinecke committed
88
89
from .logger import logger

Martin Reinecke's avatar
Martin Reinecke committed
90
from .linearization import Linearization
91

92
93
from .operator_spectrum import operator_spectrum

94
# We deliberately don't set __all__ here, because we don't want people to do a
Martin Reinecke's avatar
5->6    
Martin Reinecke committed
95
# "from nifty6 import *"; that would swamp the global namespace.