perf(demo): avoid some GC but not all

PoisotLab · Dec 20, 2024 · bf15015 · bf15015
1 parent 7c1c3a0
commit bf15015
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/SDeMo/src/explanations/shapley.jl b/SDeMo/src/explanations/shapley.jl
@@ -69,7 +69,9 @@ function explain(
     samples = 200,
     kwargs...,
 ) where {T <: AbstractSDM}
-    predictor = (x) -> predict(model, x; kwargs...)
+    function predictor(x)
+        return predict(model, x; kwargs...)
+    end
     instances = isnothing(instances) ? features(model) : instances
     if isnothing(observation)
         return shap_all_points(predictor, instances, features(model), j, samples)