Brain graph plot updated

berenslab · Oct 17, 2024 · 2c44b6a · 2c44b6a
1 parent f32f741
commit 2c44b6a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/retinal_rl/analysis/plot.py b/retinal_rl/analysis/plot.py
@@ -223,6 +223,7 @@ def plot_brain_and_optimizers(brain: Brain, objective: Objective[ContextT]) -> F
 
     plt.title("Brain Connectome and Loss Targets")
     plt.tight_layout()
+    plt.axis("equal")
     plt.axis("off")
 
     return fig