added explain_func_kwargs to custom_preprocess

understandable-machine-intelligence-lab · Dec 4, 2023 · c40d1c6 · c40d1c6
1 parent 65c0130
commit c40d1c6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/quantus/metrics/randomisation/smooth_mprt.py b/quantus/metrics/randomisation/smooth_mprt.py
@@ -546,7 +546,9 @@ def custom_preprocess(
             return None
 
         a_batch_chunks = []
-        for a_chunk in self.generate_explanations(model, x_batch, y_batch):
+        for a_chunk in self.generate_explanations(
+            model, x_batch, y_batch, **{**kwargs, **self.explain_func_kwargs}
+        ):
             a_batch_chunks.extend(a_chunk)
         return dict(a_batch=np.asarray(a_batch_chunks))