aboutsummaryrefslogtreecommitdiff
path: root/Examples/Tests/PythonWrappers
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Tests/PythonWrappers')
-rwxr-xr-xExamples/Tests/PythonWrappers/PICMI_inputs_2d.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/Examples/Tests/PythonWrappers/PICMI_inputs_2d.py b/Examples/Tests/PythonWrappers/PICMI_inputs_2d.py
index 8e4648554..597bcd69a 100755
--- a/Examples/Tests/PythonWrappers/PICMI_inputs_2d.py
+++ b/Examples/Tests/PythonWrappers/PICMI_inputs_2d.py
@@ -1,8 +1,8 @@
#!/usr/bin/env python3
-import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1.axes_divider import make_axes_locatable
+import numpy as np
from pywarpx import picmi
# Number of time steps
@@ -171,6 +171,7 @@ sim.initialize_warpx()
# Get fields data using Python wrappers
import pywarpx.fields as pwxf
+
Ex = pwxf.ExFPWrapper(include_ghosts = include_ghosts)
Ey = pwxf.EyFPWrapper(include_ghosts = include_ghosts)
Ez = pwxf.EzFPWrapper(include_ghosts = include_ghosts)