Run pre-commit

vanderschaarlab · Dec 2, 2024 · 1ad7858 · 1ad7858
1 parent 575a0ad
commit 1ad7858
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autoprognosis/plugins/explainers/plugin_kernel_shap.py b/src/autoprognosis/plugins/explainers/plugin_kernel_shap.py
@@ -128,7 +128,7 @@ def explain(self, X: pd.DataFrame) -> np.ndarray:
         X = pd.DataFrame(X, columns=self.feature_names)
         importance = np.asarray(self.explainer.shap_values(X))
         if self.task_type == "classification":
-            importance = importance[:,:,1]
+            importance = importance[:, :, 1]
 
         return importance