Added compat to improve kerneldiagmatrix

Author: theogf

Project.toml

@@ -3,6 +3,7 @@ uuid = "ec8451be-7e33-11e9-00cf-bbf324bd1392"
 version = "0.2.2"
 
 [deps]
+Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
 Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 PDMats = "90014a1f-27ba-587c-ab20-58faa44d9150"
@@ -11,6 +12,7 @@ StatsFuns = "4c63d2b9-4356-54db-8cca-17b64c39e42c"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
+Compat = "2.2"
 Distances = "0.8"
 PDMats = "0.9"
 SpecialFunctions = "0"

src/KernelFunctions.jl

@@ -14,6 +14,7 @@ export KernelSum, KernelProduct
 
 export Transform, SelectTransform, ChainTransform, ScaleTransform, LowRankTransform, IdentityTransform, FunctionTransform
 
+using Compat
 using Distances, LinearAlgebra
 using SpecialFunctions: lgamma, besselk
 using Zygote: @adjoint

src/matrix/kernelmatrix.jl

@@ -108,9 +108,9 @@ function kerneldiagmatrix(
         )
         @assert obsdim ∈ [1,2] "obsdim should be 1 or 2 (see docs of kernelmatrix))"
         if obsdim == 1
-            [@views _kernel(κ,X[i,:],X[i,:]) for i in 1:size(X,obsdim)]
+            @compat eachrow(X) .|> x->_kernel(κ,x,x) #[@views _kernel(κ,X[i,:],X[i,:]) for i in 1:size(X,obsdim)]
         elseif obsdim == 2
-            [@views _kernel(κ,X[:,i],X[:,i]) for i in 1:size(X,obsdim)]
+            @compat eachcol(X) .|> x->_kernel(κ,x,x) #[@views _kernel(κ,X[:,i],X[:,i]) for i in 1:size(X,obsdim)]
         end
 end