Commit 1ac5f37d authored by Lukas Platz's avatar Lukas Platz

hide SphericalConvolutionOperator

parent 360cac16
Pipeline #43918 passed with stages
in 8 minutes and 8 seconds
...@@ -50,8 +50,7 @@ from .operators.value_inserter import ValueInserter ...@@ -50,8 +50,7 @@ from .operators.value_inserter import ValueInserter
from .operators.energy_operators import ( from .operators.energy_operators import (
EnergyOperator, GaussianEnergy, PoissonianEnergy, InverseGammaLikelihood, EnergyOperator, GaussianEnergy, PoissonianEnergy, InverseGammaLikelihood,
BernoulliEnergy, StandardHamiltonian, AveragedEnergy) BernoulliEnergy, StandardHamiltonian, AveragedEnergy)
from .operators.convolution_operators import ( from .operators.convolution_operators import SphericalFuncConvolutionOperator
SphericalConvolutionOperator, SphericalFuncConvolutionOperator)
from .probing import probe_with_posterior_samples, probe_diagonal, \ from .probing import probe_with_posterior_samples, probe_diagonal, \
StatCalculator StatCalculator
...@@ -44,10 +44,10 @@ def SphericalFuncConvolutionOperator(domain, func): ...@@ -44,10 +44,10 @@ def SphericalFuncConvolutionOperator(domain, func):
if not isinstance(domain[0], (HPSpace, GLSpace)): if not isinstance(domain[0], (HPSpace, GLSpace)):
raise TypeError("need a spherical domain") raise TypeError("need a spherical domain")
kernel = domain[0].get_default_codomain().get_conv_kernel_from_func(func) kernel = domain[0].get_default_codomain().get_conv_kernel_from_func(func)
return SphericalConvolutionOperator(domain, kernel) return _SphericalConvolutionOperator(domain, kernel)
class SphericalConvolutionOperator(EndomorphicOperator): class _SphericalConvolutionOperator(EndomorphicOperator):
"""Convolves with kernel living on the appropriate LMSpace""" """Convolves with kernel living on the appropriate LMSpace"""
def __init__(self, domain, kernel): def __init__(self, domain, kernel):
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