Apply suggestions from code review

Evovest · Oct 7, 2023 · adfacca · adfacca
1 parent 1c391a8
commit adfacca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/EvoTreesCUDAExt/predict.jl b/ext/EvoTreesCUDAExt/predict.jl
@@ -102,7 +102,7 @@ function predict_kernel!(
 end
 
 # prediction from single tree - assign each observation to its final leaf
-function predict!(
+function EvoTrees.predict!(
     pred::CuMatrix{T},
     tree::EvoTrees.Tree{L,K},
     x_bin::CuMatrix,
@@ -125,7 +125,7 @@ function predict!(
     CUDA.synchronize()
 end
 
-function predict!(
+function EvoTrees.predict!(
     pred::CuMatrix{T},
     tree::EvoTrees.Tree{L,K},
     x_bin::CuMatrix,