come on

src/counterfactuals/flatten.jl

@@ -51,4 +51,4 @@ Returns the encoded representation of `flat_ce.target`.
 """
 function target_encoded(flat_ce::FlattenedCE, data::CounterfactualData)
     return data.output_encoder(flat_ce.target; y_levels=data.y_levels)
-end
+end

src/counterfactuals/utils.jl

@@ -96,4 +96,5 @@ end
 
 Overloads the function for [`CounterfactualExplanation`](@ref) to use the counterfactual data's labels if no data is provided.
 """
-find_potential_neighbours(ce::CounterfactualExplanation, n::Int=1000) = find_potential_neighbours(ce, ce.data, n)
+find_potential_neighbours(ce::CounterfactualExplanation, n::Int=1000) =
+    find_potential_neighbours(ce, ce.data, n)

src/objectives/penalties.jl

@@ -121,11 +121,13 @@ function distance_from_target(
 )
 
     # Get potential neighbours:
-    get!(
-        ce.search,
-        :potential_neighbours,
-        CounterfactualExplanations.find_potential_neighbours(ce, K),
-    )
+    ChainRulesCore.ignore_derivatives() do
+        get!(
+            ce.search,
+            :potential_neighbours,
+            CounterfactualExplanations.find_potential_neighbours(ce, K),
+        )
+    end
 
     ys = ce.search[:potential_neighbours]
     if K > size(ys, 2)