bug fix in scatter plot

bug fix in ebs_plot -> fix_collinear_spin
romerogroup · Jan 11, 2024 · eebd36f · eebd36f
1 parent 9a2f75d
commit eebd36f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyprocar/core/ebs.py b/pyprocar/core/ebs.py
@@ -1133,6 +1133,7 @@ def fix_collinear_spin(self):
 
         if self.projected is not None:
             print('old projected.shape', self.projected.shape)
+            self.projected[... , -1] = -self.projected[... , -1]
             shape = list(self.projected.shape)
             shape[1] = shape[1]*2
             shape[-1] = 1

diff --git a/pyprocar/plotter/ebs_plot.py b/pyprocar/plotter/ebs_plot.py
@@ -242,7 +242,7 @@ def plot_scatter(self,
                         alpha=self.config['opacity']['value'][ispin],
                     )
         if self.config['plot_color_bar']['value'] and color_weights is not None:
-            self.cb = self.fig.colorbar(lc, ax=self.ax)
+            self.cb = self.fig.colorbar(sc, ax=self.ax)
 
     def plot_parameteric(
         self,