MAINT: upright units

ChristopherMayes · Dec 17, 2024 · e28cc13 · e28cc13
1 parent 657b384
commit e28cc13
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pmd_beamphysics/wavefront.py b/pmd_beamphysics/wavefront.py
@@ -1492,13 +1492,13 @@ def plot(dat, title: str):
             if rspace:
                 # See tech note 3.2
                 power_density = 1 / 1e4 * np.abs(mesh_data) ** 2 / (2.0 * Z0)
-                img = plot(power_density, "Power density of mesh $W/cm^2$")
+                img = plot(power_density, "Power density of mesh (W/cm$^2$)")
             else:
                 # See tech note 3.6 (coefficient is in 3.4)
                 power_density = (
                     np.abs(mesh_data) ** 2 / (2.0 * Z0) * self.fft_unit_coeff
                 )
-                img = plot(power_density, "Power density of mesh $J/eV/rad^2$")
+                img = plot(power_density, "Power density of mesh (J/eV/rad$^2$)")
 
         elif key == "phase":
             phase = np.angle(mesh_data)