__init__.py 4.69 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
Philipp Arras's avatar
Philipp Arras committed
22
from .operators.adder import Adder
23
from .operators.log1p import Log1p
Martin Reinecke's avatar
Martin Reinecke committed
24
from .operators.diagonal_operator import DiagonalOperator
Martin Reinecke's avatar
cleanup  
Martin Reinecke committed
25
from .operators.distributors import DOFDistributor, PowerDistributor
Philipp Arras's avatar
Philipp Arras committed
26
from .operators.domain_tuple_field_inserter import DomainTupleFieldInserter
27
from .operators.contraction_operator import ContractionOperator
28
from .operators.linear_interpolation import LinearInterpolator
Martin Reinecke's avatar
Martin Reinecke committed
29
30
from .operators.endomorphic_operator import EndomorphicOperator
from .operators.exp_transform import ExpTransform
Martin Reinecke's avatar
Martin Reinecke committed
31
32
33
from .operators.harmonic_operators import (
    FFTOperator, HartleyOperator, SHTOperator, HarmonicTransformOperator,
    HarmonicSmoothingOperator)
Martin Reinecke's avatar
Martin Reinecke committed
34
from .operators.field_zero_padder import FieldZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
35
36
37
38
from .operators.inversion_enabler import InversionEnabler
from .operators.linear_operator import LinearOperator
from .operators.mask_operator import MaskOperator
from .operators.qht_operator import QHTOperator
39
from .operators.regridding_operator import RegriddingOperator
Martin Reinecke's avatar
Martin Reinecke committed
40
41
42
43
44
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
45
from .operators.block_diagonal_operator import BlockDiagonalOperator
46
from .operators.outer_product_operator import OuterProduct
Martin Reinecke's avatar
Martin Reinecke committed
47
from .operators.simple_linear_operators import (
48
    VdotOperator, ConjugationOperator, Realizer,
49
    FieldAdapter, ducktape, GeometryRemover, NullOperator,
Philipp Arras's avatar
Philipp Arras committed
50
    MatrixProductOperator, PartialExtractor)
Philipp Arras's avatar
Philipp Arras committed
51
from .operators.value_inserter import ValueInserter
Martin Reinecke's avatar
Martin Reinecke committed
52
from .operators.energy_operators import (
Martin Reinecke's avatar
merge  
Martin Reinecke committed
53
    EnergyOperator, GaussianEnergy, PoissonianEnergy, InverseGammaLikelihood,
54
    BernoulliEnergy, StandardHamiltonian, AveragedEnergy, QuadraticFormOperator,
55
    Squared2NormOperator, StudentTEnergy)
56
from .operators.convolution_operators import FuncConvolutionOperator
Martin Reinecke's avatar
Martin Reinecke committed
57

Martin Reinecke's avatar
Martin Reinecke committed
58
from .probing import probe_with_posterior_samples, probe_diagonal, \
59
    StatCalculator, approximation2endo
Martin Reinecke's avatar
Martin Reinecke committed
60
61

from .minimization.line_search import LineSearch
Martin Reinecke's avatar
Martin Reinecke committed
62
from .minimization.iteration_controllers import (
Martin Reinecke's avatar
Martin Reinecke committed
63
    IterationController, GradientNormController, DeltaEnergyController,
64
    GradInfNormController, AbsDeltaEnergyController)
Martin Reinecke's avatar
Martin Reinecke committed
65
66
67
from .minimization.minimizer import Minimizer
from .minimization.conjugate_gradient import ConjugateGradient
from .minimization.nonlinear_cg import NonlinearCG
68
from .minimization.descent_minimizers import (
69
70
    DescentMinimizer, SteepestDescent, VL_BFGS, L_BFGS, RelaxedNewton,
    NewtonCG)
Martin Reinecke's avatar
Martin Reinecke committed
71
from .minimization.scipy_minimizer import L_BFGS_B
Martin Reinecke's avatar
Martin Reinecke committed
72
73
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
Martin Reinecke's avatar
Martin Reinecke committed
74
from .minimization.energy_adapter import EnergyAdapter
75
from .minimization.metric_gaussian_kl import MetricGaussianKL
Martin Reinecke's avatar
Martin Reinecke committed
76

Martin Reinecke's avatar
Martin Reinecke committed
77
from .sugar import *
78
from .plot import Plot
Philipp Arras's avatar
Fixups  
Philipp Arras committed
79

Martin Reinecke's avatar
Martin Reinecke committed
80
81
from .library.smooth_linear_amplitude import (
    SLAmplitude, LinearSLAmplitude, CepstrumOperator)
Philipp Arras's avatar
Philipp Arras committed
82
from .library.inverse_gamma_operator import InverseGammaOperator
Martin Reinecke's avatar
Martin Reinecke committed
83
from .library.los_response import LOSResponse
Philipp Arras's avatar
Philipp Arras committed
84
85
from .library.dynamic_operator import (dynamic_operator,
                                       dynamic_lightcone_operator)
86
from .library.light_cone_operator import LightConeOperator
Martin Reinecke's avatar
merge  
Martin Reinecke committed
87

Philipp Arras's avatar
Fixup  
Philipp Arras committed
88
from .library.wiener_filter_curvature import WienerFilterCurvature
89
from .library.correlated_fields import CorrelatedField, MfCorrelatedField
90
from .library.adjust_variances import (make_adjust_variances_hamiltonian,
Martin Reinecke's avatar
Martin Reinecke committed
91
                                       do_adjust_variances)
Philipp Arras's avatar
Philipp Arras committed
92
from .library.gridder import GridderMaker
Martin Reinecke's avatar
Martin Reinecke committed
93

Philipp Arras's avatar
Move it  
Philipp Arras committed
94
from . import extra
95

Martin Reinecke's avatar
Martin Reinecke committed
96
from .utilities import memo, frozendict
Martin Reinecke's avatar
Martin Reinecke committed
97

Martin Reinecke's avatar
Martin Reinecke committed
98
99
from .logger import logger

Martin Reinecke's avatar
Martin Reinecke committed
100
from .linearization import Linearization
101

102
103
from .operator_spectrum import operator_spectrum

Martin Reinecke's avatar
Martin Reinecke committed
104
105
from . import internal_config
_scheme = internal_config.parallelization_scheme()
Martin Reinecke's avatar
Martin Reinecke committed
106
107
108
if _scheme == "Samples":
    from .minimization.metric_gaussian_kl_mpi import MetricGaussianKL_MPI

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