From 70bff18f4f3ed284cb1f7541a8d4246803cdf1db Mon Sep 17 00:00:00 2001 From: Meganton <43831195+Meganton@users.noreply.github.com> Date: Wed, 5 Feb 2025 16:26:27 +0100 Subject: [PATCH 1/2] Fix error in trajectory_plotting.py removed titleside='right', to prevent error --- tuned_lens/plotting/trajectory_plotting.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tuned_lens/plotting/trajectory_plotting.py b/tuned_lens/plotting/trajectory_plotting.py index 1b89e83..de842a4 100644 --- a/tuned_lens/plotting/trajectory_plotting.py +++ b/tuned_lens/plotting/trajectory_plotting.py @@ -178,8 +178,7 @@ def heatmap( y=self._layer_labels, z=self.stats if not log_scale else np.log10(self.stats), colorbar=dict( - title=f"{self.name} ({self.units})", - titleside="right", + title=f"{self.name} ({self.units})" ), colorscale=colorscale, zmax=max if not log_scale else np.log10(max), From d0e363076726fe99a9d2b6a72b74e9cea2f2abdf Mon Sep 17 00:00:00 2001 From: Meganton Date: Wed, 5 Feb 2025 16:29:03 +0100 Subject: [PATCH 2/2] removed titleside='right' to prevent error when executing --- tuned_lens/plotting/trajectory_plotting.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tuned_lens/plotting/trajectory_plotting.py b/tuned_lens/plotting/trajectory_plotting.py index 1b89e83..07dbe37 100644 --- a/tuned_lens/plotting/trajectory_plotting.py +++ b/tuned_lens/plotting/trajectory_plotting.py @@ -179,7 +179,6 @@ def heatmap( z=self.stats if not log_scale else np.log10(self.stats), colorbar=dict( title=f"{self.name} ({self.units})", - titleside="right", ), colorscale=colorscale, zmax=max if not log_scale else np.log10(max),