Update _plot_pressure.py

PrincetonUniversity · Nov 18, 2024 · e406819 · e406819
1 parent 6d2693c
commit e406819
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Utilities/pythontools/py_spec/output/_plot_pressure.py b/Utilities/pythontools/py_spec/output/_plot_pressure.py
@@ -10,7 +10,7 @@ def plot_pressure(self, normalize=True, ax=None, **kwargs):
     import matplotlib.pyplot as plt
 
     pressure = np.atleast_1d(self.input.physics.pressure) * self.input.physics.pscale
-    tflux = self.output.tflux[: len(pressure)]
+    tflux = np.atleast_1d(self.output.tflux)[: len(pressure)]
     if not normalize:
         #  remove  mu_0
         pressure /= 4 * np.pi * 1.0e-7