Update utils.py

Smartappli · May 25, 2024 · 5cd303a · 5cd303a
1 parent ce237e8
commit 5cd303a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fl_common/models/utils.py b/fl_common/models/utils.py
@@ -461,7 +461,7 @@ def generate_xai_heatmaps(model, image_tensor, label, save_dir, methods=None):
         plt.imshow(attributions_np, cmap="viridis")
         plt.title(f"XAI Heatmap for {method_name} (Label: {label})")
         plt.colorbar()
-        save_path = save_dir, +"/" + f"xai_heatmap_{method_name}_{label}.png"
+        save_path = save_dir, + '/' + f"xai_heatmap_{method_name}_{label}.png"
         plt.savefig(save_path)
         plt.show()