aboutsummaryrefslogtreecommitdiff
path: root/Examples/Modules/laser_injection/analysis_2d.py
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Modules/laser_injection/analysis_2d.py')
-rwxr-xr-xExamples/Modules/laser_injection/analysis_2d.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/Examples/Modules/laser_injection/analysis_2d.py b/Examples/Modules/laser_injection/analysis_2d.py
index e26ae4478..377931718 100755
--- a/Examples/Modules/laser_injection/analysis_2d.py
+++ b/Examples/Modules/laser_injection/analysis_2d.py
@@ -17,15 +17,18 @@
# the simulation and it compares it with theory. It also checks that the
# central frequency of the Fourier transform is the expected one.
-import yt
+import os
import sys
+
import matplotlib
+
matplotlib.use('Agg')
import matplotlib.pyplot as plt
+from mpl_toolkits.axes_grid1 import make_axes_locatable
import numpy as np
-import os
from scipy.signal import hilbert
-from mpl_toolkits.axes_grid1 import make_axes_locatable
+import yt
+
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI