fix fallbacks for sparse argmax and soft argmax regularizations

Author: BatyLeo

src/regularized/soft_argmax.jl

@@ -16,5 +16,5 @@ function soft_argmax(z::AbstractVector)
 end
 
 function soft_argmax_regularization(y::AbstractVector)
-    return isprobadist(y) ? negative_shannon_entropy(y) : typemax(R)
+    return isprobadist(y) ? negative_shannon_entropy(y) : typemax(eltype(y))
 end

src/regularized/sparse_argmax.jl

@@ -16,7 +16,7 @@ function sparse_argmax(z::AbstractVector; kwargs...)
 end
 
 function sparse_argmax_regularization(y::AbstractVector)
-    return isprobadist(y) ? half_square_norm(y) : typemax(R)
+    return isprobadist(y) ? half_square_norm(y) : typemax(eltype(y))
 end
 
 """