diff --git a/src/Models.jl b/src/Models.jl index 5ae43d7..816c8df 100644 --- a/src/Models.jl +++ b/src/Models.jl @@ -44,7 +44,7 @@ Predict targets for the provided the collection of `inputs` and [`Model`](@ref). If the `estimate_type(model)` is [`PointEstimate`](@ref) then this function should return another `AbstractMatrix` in which each column contains the prediction for a single input. -If the `estimate_type(model) == [`DistributionEstimate`](@ref) then this function should +If the `estimate_type(model)` is [`DistributionEstimate`](@ref) then this function should return a `AbstractVector{<:Distribution}`. """ function predict end