Return only the value of evaluate when possible

OlivierHnt · Aug 28, 2023 · 611b2fc · 611b2fc
1 parent c9ccf79
commit 611b2fc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/sequence_spaces/special_operators/evaluation.jl b/src/sequence_spaces/special_operators/evaluation.jl
@@ -84,9 +84,13 @@ function evaluate(a::Sequence, x)
     CoefType = _coeftype(ℰ, space_a, eltype(a))
     c = Sequence(new_space, Vector{CoefType}(undef, dimension(new_space)))
     _apply!(c, ℰ, a)
-    return c
+    return _return_evaluate(c, x)
 end
 
+_return_evaluate(a::Sequence, ::Any) = a
+_return_evaluate(a::Sequence{<:BaseSpace}, ::Number) = coefficients(a)[1]
+_return_evaluate(a::Sequence{TensorSpace{T}}, ::NTuple{N,Number}) where {N,T<:NTuple{N,BaseSpace}} = coefficients(a)[1]
+
 """
     evaluate!(c::Sequence, a::Sequence, x)