From bb087aa80ae248ceb7f22bab10c0501ef28f97d7 Mon Sep 17 00:00:00 2001
From: Dehann Fourie <6412556+dehann@users.noreply.github.com>
Date: Mon, 20 Nov 2023 04:13:27 -0800
Subject: [PATCH] Update DualTree01.jl

---
 src/DualTree01.jl | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/DualTree01.jl b/src/DualTree01.jl
index c83b7b6..4cdb242 100644
--- a/src/DualTree01.jl
+++ b/src/DualTree01.jl
@@ -428,21 +428,21 @@ Evaluate the KDE object at given points.
 
 > **Note**, must use Array{Float64,2} when passing in evaluation points.
 """
-function (bd::BallTreeDensity)(pos::Array{Float64,2},
+function (bd::BallTreeDensity)(pos::AbstractMatrix{Float64},
                                lvFlag::Bool=false,
                                errTol::Float64=1e-3,
                                addop=(+,),
                                diffop=(-,) )
   evaluateDualTree(bd, pos, lvFlag, errTol, addop, diffop)
 end
-function (bd::BallTreeDensity)(pos::Array{Float64,1},
+function (bd::BallTreeDensity)(pos::AbstractVector{Float64},
                                lvFlag::Bool=false,
                                errTol::Float64=1e-3,
                                addop=(+,),
                                diffop=(-,) )
   # TODO should it not be reshape(pos,1,:) instead?
   # @warn "whoa! check reshape inside this eval balltree function"
-  evaluateDualTree(bd, reshape(pos,1,:), lvFlag, errTol, addop, diffop)
+  evaluateDualTree(bd, reshape(pos,:,1), lvFlag, errTol, addop, diffop)
 end