Skip to content

Commit 860a4c7

Browse files
committed
Relaxed transform to be used on Kernel
1 parent d6dfdd3 commit 860a4c7

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

src/kernels/transformedkernel.jl

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -31,21 +31,27 @@ _scale(t::ScaleTransform, metric, x, y) = evaluate(metric, t(x), t(y))
3131

3232
"""
3333
```julia
34-
transform(k::BaseKernel, t::Transform) (1)
35-
transform(k::BaseKernel, ρ::Real) (2)
36-
transform(k::BaseKernel, ρ::AbstractVector) (3)
34+
transform(k::Kernel, t::Transform) (1)
35+
transform(k::Kernel, ρ::Real) (2)
36+
transform(k::Kernel, ρ::AbstractVector) (3)
3737
```
3838
(1) Create a TransformedKernel with transform `t` and kernel `k`
3939
(2) Same as (1) with a `ScaleTransform` with scale `ρ`
4040
(3) Same as (1) with an `ARDTransform` with scales `ρ`
4141
"""
4242
transform
4343

44-
transform(k::BaseKernel, t::Transform) = TransformedKernel(k, t)
44+
transform(k::Kernel, t::Transform) = TransformedKernel(k, t)
4545

46-
transform(k::BaseKernel, ρ::Real) = TransformedKernel(k, ScaleTransform(ρ))
46+
transform(k::TransformedKernel, t::Transform) =
47+
TransformedKernel(k.kernel, t k.transform)
48+
49+
transform(k::Kernel, ρ::Real) = transform(k, ScaleTransform(ρ))
50+
51+
transform(k::Kernel, ρ::AbstractVector) = transform(k, ARDTransform(ρ))
52+
53+
transform(k::Kernel, ::Nothing) = k
4754

48-
transform(k::BaseKernel::AbstractVector) = TransformedKernel(k, ARDTransform(ρ))
4955

5056
kernel(κ) = κ.kernel
5157

0 commit comments

Comments
 (0)