[fes] remove units from plot

luigibonati · Mar 27, 2022 · 6567fb0 · 6567fb0
1 parent d1616c3
commit 6567fb0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlcvs/utils/fes.py b/mlcvs/utils/fes.py
@@ -171,14 +171,14 @@ def compute_fes(X, temp=300, kbt=None, num_samples=100, bounds=None, bandwidth=0
                 ax.errorbar(grid,fes2,error)
             else:
                 ax.plot(grid,fes2)  
-            ax.set_ylabel('FES [kJ/mol]')
+            ax.set_ylabel('FES')
         elif dim == 2: 
             fes2 = np.copy(O)
             if plot_max_fes is not None:
                 fes2[fes2>plot_max_fes] = np.nan
             extent = [item for sublist in bounds for item in sublist] 
             pp = ax.contourf(fes2,cmap='fessa',extent=extent)#,vmax=max_fes)
             cbar = plt.colorbar(pp,ax=ax)
-            cbar.set_label('FES [kJ/mol]')
+            cbar.set_label('FES')
 
     return fes,grid,bounds,error