Add support for nothing

I have been using this quite often to conditionally drop specific parameters.
JuliaGaussianProcesses · Feb 5, 2024 · 6894f28 · 6894f28
1 parent 1fdb2cb
commit 6894f28
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/flatten.jl b/src/flatten.jl
@@ -19,6 +19,12 @@ function flatten end
 
 flatten(x) = flatten(Float64, x)
 
+function ParameterHandling.flatten(::Type{T}, ::Nothing) where {T<:Real} 
+    v = T[]
+    unflatten_to_Nothing(::Vector{T}) = nothing
+    return v, unflatten_to_Nothing
+end
+
 function flatten(::Type{T}, x::Integer) where {T<:Real}
     v = T[]
     unflatten_to_Integer(v::Vector{T}) = x