Improve return_type method

[artememelin]

Pull request checklist

• Did you bump the project version?
• Did you add a description to the Pull Request?
• Did you add new tests?
• Did you add reviewers?

[artememelin]

using Dates
using TimeArrays

struct Trade
    price::Float64
    volume::Float64
end

function w_avg(p::AbstractVector{Trade})
    return sum(x -> x.price * x.volume, p) / sum(x -> x.volume, p)
end

function TimeArrays.return_type(::typeof(w_avg), ::Type{T}) where {T<:Trade}
    return Float64
end

t_array = TimeArray([
   TimeTick(DateTime("2024-01-02"), Trade(1.0, 10)),
   TimeTick(DateTime("2024-01-03"), Trade(2.0, 20)),
   TimeTick(DateTime("2024-01-05"), Trade(3.0, 30)),
   TimeTick(DateTime("2024-01-06"), Trade(4.0, 40)),
   TimeTick(DateTime("2024-01-09"), Trade(5.0, 50)),
])

ta_rolling(w_avg, t_array, Day(3))

[github-actions]

📘 A preview of the documentation will be here soon