Changed the approach for the relaxation of validate_inputs

Author: theogf

src/utils.jl

@@ -20,9 +20,6 @@ function vec_of_vecs(X::AbstractMatrix; obsdim::Int = 2)
     end
 end
 
-dim(x::AbstractVector{<:AbstractVector{<:Real}}) = length(first(x))
-dim(x::AbstractVector{Tuple{Any,Int}}) = 1
-
 """
     ColVecs(X::AbstractMatrix)
 
@@ -94,6 +91,21 @@ For a transform return its parameters, for a `ChainTransform` return a vector of
 """
 #params
 
+dim(x) = 0
+dim(x::AbstractVector{<:AbstractVector{<:Real}}) = length(first(x))
+dim(x::AbstractVector{<:Real}) = 1
+dim(x::AbstractVector{Tuple{Any,Int}}) = 1
+
+
+function validate_inputs(x, y)
+    if dim(x) != dim(y) # Passes by default if `dim` is not defined
+        throw(DimensionMismatch(
+            "Dimensionality of x ($(dim(x))) not equality to that of y ($(dim(y)))",
+        ))
+    end
+    return nothing
+end
+
 
 function validate_inplace_dims(K::AbstractMatrix, x::AbstractVector, y::AbstractVector)
     validate_inputs(x, y)
@@ -117,13 +129,3 @@ function validate_inplace_dims(K::AbstractVector, x::AbstractVector)
         ))
     end
 end
-
-validate_inputs(x, y) = nothing
-
-function validate_inputs(x::V, y::V) where {V<:Union{RowVecs, ColVecs, AbstractVector{<:AbstractVector{<:Real}}}}
-    if dim(x) != dim(y)
-        throw(DimensionMismatch(
-            "Dimensionality of x ($(dim(x))) not equality to that of y ($(dim(y)))",
-        ))
-    end
-end

test/matrix/kernelmatrix.jl

@@ -90,7 +90,7 @@ KernelFunctions.kappa(::ToySimpleKernel, d) = exp(-d / 2)
 
             tmp_diag = Vector{Float64}(undef, length(x))
             @test kerneldiagmatrix!(tmp_diag, k, x) ≈ kerneldiagmatrix(k, x)
-            @test_throws DimensionMismatch kerneldiagmatrix!(tmp_diag, k, y)            
+            @test_throws DimensionMismatch kerneldiagmatrix!(tmp_diag, k, y)
         end
     end