Fixed a dispatch bug that would break derivatives of R->R^n functions

JuliaDiff · Nov 20, 2017 · 5b65ff8 · 5b65ff8
1 parent a2466d9
commit 5b65ff8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/derivatives.jl b/src/derivatives.jl
@@ -2,15 +2,15 @@
 Compute the derivative df of a callable f on a collection of points x.
 Generic fallbacks for AbstractArrays that are not StridedArrays.
 =#
-function finite_difference(f, x::AbstractArray{<:Number},
+function finite_difference(f, x::Union{<:Number,AbstractArray{<:Number}},
     fdtype::DataType=Val{:central}, funtype::DataType=Val{:Real}, wrappertype::DataType=Val{:Default},
     fx::Union{Void,AbstractArray{<:Number}}=nothing, epsilon::Union{Void,AbstractArray{<:Real}}=nothing, return_type::DataType=eltype(x))
 
     df = zeros(return_type, size(x))
     finite_difference!(df, f, x, fdtype, funtype, wrappertype, fx, epsilon, return_type)
 end
 
-function finite_difference!(df::AbstractArray{<:Number}, f, x::AbstractArray{<:Number},
+function finite_difference!(df::AbstractArray{<:Number}, f, x::Union{<:Number,AbstractArray{<:Number}},
     fdtype::DataType=Val{:central}, funtype::DataType=Val{:Real}, wrappertype::DataType=Val{:Default},
     fx::Union{Void,AbstractArray{<:Number}}=nothing, epsilon::Union{Void,AbstractArray{<:Real}}=nothing, return_type::DataType=eltype(x))