From 4195b9f0fdf4eec4c6fcad72b997cbafef81178a Mon Sep 17 00:00:00 2001
From: AlQuemist <alquemist@Lyriks>
Date: Wed, 14 Aug 2024 14:25:12 +0200
Subject: [PATCH] rawEx: import matplotlib.pyplot when needed

---
 rawEx/ff/BarLorentz.py                                 | 1 +
 rawEx/ff/Bipyramid4.py                                 | 1 +
 rawEx/ff/Box.py                                        | 1 +
 rawEx/ff/CantellatedCube.py                            | 1 +
 rawEx/ff/Cone.py                                       | 1 +
 rawEx/ff/CosineRippleBox.py                            | 1 +
 rawEx/ff/Cylinder.py                                   | 1 +
 rawEx/ff/Dodecahedron.py                               | 1 +
 rawEx/ff/EllipsoidalCylinder.py                        | 1 +
 rawEx/ff/HemiEllipsoid.py                              | 1 +
 rawEx/ff/HorizontalCylinder.py                         | 1 +
 rawEx/ff/Icosahedron.py                                | 1 +
 rawEx/ff/PlatonicOctahedron.py                         | 1 +
 rawEx/ff/PlatonicTetrahedron.py                        | 1 +
 rawEx/ff/Prism3.py                                     | 1 +
 rawEx/ff/Prism6.py                                     | 1 +
 rawEx/ff/Pyramid2.py                                   | 1 +
 rawEx/ff/Pyramid3.py                                   | 1 +
 rawEx/ff/Pyramid4.py                                   | 1 +
 rawEx/ff/Pyramid6.py                                   | 1 +
 rawEx/ff/SawtoothRippleBox.py                          | 1 +
 rawEx/ff/Sphere.py                                     | 1 +
 rawEx/ff/Spheroid.py                                   | 1 +
 rawEx/ff/TruncatedCube.py                              | 1 +
 rawEx/ff/TruncatedSphere.py                            | 1 +
 rawEx/ff/TruncatedSpheroid.py                          | 1 +
 rawEx/offspec/Offspec1.py                              | 1 +
 rawEx/offspec/OffspecLambda.py                         | 1 +
 rawEx/offspec/OffspecResolved.py                       | 1 +
 rawEx/scatter2d/ApproximationDA.py                     | 1 +
 rawEx/scatter2d/ApproximationLMA.py                    | 1 +
 rawEx/scatter2d/ApproximationSSCA.py                   | 2 +-
 rawEx/scatter2d/BeamDivergence.py                      | 1 +
 rawEx/scatter2d/BiMaterialCylinders.py                 | 1 +
 rawEx/scatter2d/BoxesWithSpecularPeak.py               | 1 +
 rawEx/scatter2d/BroadSpecular.py                       | 2 ++
 rawEx/scatter2d/ConstantBackground.py                  | 1 +
 rawEx/scatter2d/CoreShellNanoparticles.py              | 1 +
 rawEx/scatter2d/CoreShellNanoparticles2.py             | 1 +
 rawEx/scatter2d/CorrelatedRoughness.py                 | 1 +
 rawEx/scatter2d/CosineRipplesAtRectLattice.py          | 1 +
 rawEx/scatter2d/CustomFormFactor.py                    | 1 +
 rawEx/scatter2d/Cylinders.py                           | 1 +
 rawEx/scatter2d/CylindersAndPrisms.py                  | 2 ++
 rawEx/scatter2d/CylindersInAverageLayer.py             | 1 +
 rawEx/scatter2d/CylindersInBA.py                       | 1 +
 rawEx/scatter2d/DodecahedraSAS.py                      | 1 +
 rawEx/scatter2d/HalfSpheresInAverageTopLayer.py        | 1 +
 rawEx/scatter2d/HexagonalLatticesWithBasis.py          | 1 +
 rawEx/scatter2d/Interference1DLattice.py               | 1 +
 rawEx/scatter2d/Interference1DRadialParacrystal.py     | 1 +
 rawEx/scatter2d/Interference2DCenteredSquareLattice.py | 2 ++
 rawEx/scatter2d/Interference2DParacrystal.py           | 1 +
 rawEx/scatter2d/Interference2DRotatedSquareLattice.py  | 1 +
 rawEx/scatter2d/Interference2DSquareFiniteLattice.py   | 1 +
 rawEx/scatter2d/LatticeOrientationDistribution.py      | 1 +
 rawEx/scatter2d/MagneticCylinders1.py                  | 1 +
 rawEx/scatter2d/MagneticSpheres.py                     | 1 +
 rawEx/scatter2d/Mesocrystal.py                         | 1 +
 rawEx/scatter2d/Mesocrystal3.py                        | 1 +
 rawEx/scatter2d/Mesocrystal4.py                        | 1 +
 rawEx/scatter2d/ObsoleteSimN.py                        | 1 +
 rawEx/scatter2d/ParticleAcrossInterface.py             | 1 +
 rawEx/scatter2d/PolarizedSANS.py                       | 1 +
 rawEx/scatter2d/PolydisperseCylinders.py               | 2 ++
 rawEx/scatter2d/RectangularGrating.py                  | 1 +
 rawEx/scatter2d/Resolution.py                          | 1 +
 rawEx/scatter2d/RotatedPyramids.py                     | 1 +
 rawEx/scatter2d/RoughAndSpecular.py                    | 4 +++-
 rawEx/scatter2d/SlicedLayer.py                         | 1 +
 rawEx/scatter2d/SpheresAtHexLattice.py                 | 1 +
 rawEx/scatter2d/TriangularRipple.py                    | 1 +
 rawEx/specular/AlternatingLayers1.py                   | 1 +
 rawEx/specular/AlternatingLayers2.py                   | 1 +
 rawEx/specular/Basic.py                                | 2 ++
 rawEx/specular/BeamFullDivergence.py                   | 1 +
 rawEx/specular/Distributions.py                        | 2 ++
 rawEx/specular/MagneticLayer.py                        | 1 +
 rawEx/specular/SpecularSimulationWithRoughness.py      | 1 +
 rawEx/specular/TOFRWithResolution.py                   | 1 +
 rawEx/specular/TimeOfFlightReflectometry.py            | 1 +
 rawEx/varia/Depthprobe1.py                             | 2 +-
 rawEx/varia/OpaqueProfile1.py                          | 1 +
 rawEx/varia/Resonator.py                               | 2 ++
 rawEx/varia/TransmissionVsAlpha.py                     | 2 ++
 rawEx/varia/TransmissionVsDepth.py                     | 2 ++
 86 files changed, 97 insertions(+), 3 deletions(-)

diff --git a/rawEx/ff/BarLorentz.py b/rawEx/ff/BarLorentz.py
index 9e9d32df1d4..32d24493248 100755
--- a/rawEx/ff/BarLorentz.py
+++ b/rawEx/ff/BarLorentz.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.BarLorentz(20*nm, 6.3*nm, 4.7*nm)
diff --git a/rawEx/ff/Bipyramid4.py b/rawEx/ff/Bipyramid4.py
index 9f6330f88c7..4320e5a8b43 100755
--- a/rawEx/ff/Bipyramid4.py
+++ b/rawEx/ff/Bipyramid4.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Bipyramid4(8*nm, 4.3*nm, 0.5, 70*deg)
diff --git a/rawEx/ff/Box.py b/rawEx/ff/Box.py
index 0f6c80fc5b5..d0dc92661b4 100755
--- a/rawEx/ff/Box.py
+++ b/rawEx/ff/Box.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Box(12*nm, 4.1*nm, 6*nm)
diff --git a/rawEx/ff/CantellatedCube.py b/rawEx/ff/CantellatedCube.py
index b3220a64174..d1223ced51f 100755
--- a/rawEx/ff/CantellatedCube.py
+++ b/rawEx/ff/CantellatedCube.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.CantellatedCube(8*nm, 2.5*nm)
diff --git a/rawEx/ff/Cone.py b/rawEx/ff/Cone.py
index 0ded925d523..95c2088668d 100755
--- a/rawEx/ff/Cone.py
+++ b/rawEx/ff/Cone.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Cone(4.25*nm, 12*nm, 75*deg)
diff --git a/rawEx/ff/CosineRippleBox.py b/rawEx/ff/CosineRippleBox.py
index e960bce9410..2830df2a94c 100755
--- a/rawEx/ff/CosineRippleBox.py
+++ b/rawEx/ff/CosineRippleBox.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.CosineRippleBox(20*nm, 6.3*nm, 4.7*nm)
diff --git a/rawEx/ff/Cylinder.py b/rawEx/ff/Cylinder.py
index 5da91794d79..5752849ced0 100755
--- a/rawEx/ff/Cylinder.py
+++ b/rawEx/ff/Cylinder.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Cylinder(3.6*nm, 7.2*nm)
diff --git a/rawEx/ff/Dodecahedron.py b/rawEx/ff/Dodecahedron.py
index e7d8838e886..845baa8213a 100755
--- a/rawEx/ff/Dodecahedron.py
+++ b/rawEx/ff/Dodecahedron.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Dodecahedron(3.38*nm)
diff --git a/rawEx/ff/EllipsoidalCylinder.py b/rawEx/ff/EllipsoidalCylinder.py
index 0ab10ff6236..0c7023ce000 100755
--- a/rawEx/ff/EllipsoidalCylinder.py
+++ b/rawEx/ff/EllipsoidalCylinder.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.EllipsoidalCylinder(7.8*nm, 3*nm, 4*nm)
diff --git a/rawEx/ff/HemiEllipsoid.py b/rawEx/ff/HemiEllipsoid.py
index ec1d414c0ea..7b0e4c70dc1 100755
--- a/rawEx/ff/HemiEllipsoid.py
+++ b/rawEx/ff/HemiEllipsoid.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.HemiEllipsoid(4.4*nm, 8*nm, 4*nm)
diff --git a/rawEx/ff/HorizontalCylinder.py b/rawEx/ff/HorizontalCylinder.py
index 382fb33ffef..ef687bb12f0 100755
--- a/rawEx/ff/HorizontalCylinder.py
+++ b/rawEx/ff/HorizontalCylinder.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.HorizontalCylinder(4*nm, 9*nm, -2*nm, 2.3*nm)
diff --git a/rawEx/ff/Icosahedron.py b/rawEx/ff/Icosahedron.py
index d5a9a022c47..202777dd64a 100755
--- a/rawEx/ff/Icosahedron.py
+++ b/rawEx/ff/Icosahedron.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Icosahedron(5.13*nm)
diff --git a/rawEx/ff/PlatonicOctahedron.py b/rawEx/ff/PlatonicOctahedron.py
index d8546a7291a..c554904ff5e 100755
--- a/rawEx/ff/PlatonicOctahedron.py
+++ b/rawEx/ff/PlatonicOctahedron.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.PlatonicOctahedron(8.55*nm)
diff --git a/rawEx/ff/PlatonicTetrahedron.py b/rawEx/ff/PlatonicTetrahedron.py
index 9e3dc4391ff..acbee5ddc57 100755
--- a/rawEx/ff/PlatonicTetrahedron.py
+++ b/rawEx/ff/PlatonicTetrahedron.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.PlatonicTetrahedron(13.6*nm)
diff --git a/rawEx/ff/Prism3.py b/rawEx/ff/Prism3.py
index 3b28882aba6..71a47042282 100755
--- a/rawEx/ff/Prism3.py
+++ b/rawEx/ff/Prism3.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Prism3(9.2*nm, 8*nm)
diff --git a/rawEx/ff/Prism6.py b/rawEx/ff/Prism6.py
index bef543b8b95..1dc5adeb0fe 100755
--- a/rawEx/ff/Prism6.py
+++ b/rawEx/ff/Prism6.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Prism6(3.77*nm, 8*nm)
diff --git a/rawEx/ff/Pyramid2.py b/rawEx/ff/Pyramid2.py
index c1858a1f507..02f86d04bd8 100755
--- a/rawEx/ff/Pyramid2.py
+++ b/rawEx/ff/Pyramid2.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Pyramid2(6.4*nm, 13*nm, 7*nm, 70*deg)
diff --git a/rawEx/ff/Pyramid3.py b/rawEx/ff/Pyramid3.py
index eeb375adc76..a9fec761f69 100755
--- a/rawEx/ff/Pyramid3.py
+++ b/rawEx/ff/Pyramid3.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Pyramid3(14*nm, 6.8*nm, 70*deg)
diff --git a/rawEx/ff/Pyramid4.py b/rawEx/ff/Pyramid4.py
index efb3281622e..a87b56c3761 100755
--- a/rawEx/ff/Pyramid4.py
+++ b/rawEx/ff/Pyramid4.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Pyramid4(10*nm, 4*nm, 70*deg)
diff --git a/rawEx/ff/Pyramid6.py b/rawEx/ff/Pyramid6.py
index aec17a11e0e..085d403519b 100755
--- a/rawEx/ff/Pyramid6.py
+++ b/rawEx/ff/Pyramid6.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Pyramid6(5.43*nm, 6.8*nm, 70*deg)
diff --git a/rawEx/ff/SawtoothRippleBox.py b/rawEx/ff/SawtoothRippleBox.py
index 1270772770d..5f43b8a0baf 100755
--- a/rawEx/ff/SawtoothRippleBox.py
+++ b/rawEx/ff/SawtoothRippleBox.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.SawtoothRippleBox(20*nm, 6.3*nm, 4.7*nm, 3*nm)
diff --git a/rawEx/ff/Sphere.py b/rawEx/ff/Sphere.py
index 8f3fad7173d..0b258017825 100755
--- a/rawEx/ff/Sphere.py
+++ b/rawEx/ff/Sphere.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Sphere(4.13*nm)
diff --git a/rawEx/ff/Spheroid.py b/rawEx/ff/Spheroid.py
index 11c0862bf09..98aa57816d1 100755
--- a/rawEx/ff/Spheroid.py
+++ b/rawEx/ff/Spheroid.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.Spheroid(6.85*nm, 3*nm)
diff --git a/rawEx/ff/TruncatedCube.py b/rawEx/ff/TruncatedCube.py
index 004710c5561..c9904c74db5 100755
--- a/rawEx/ff/TruncatedCube.py
+++ b/rawEx/ff/TruncatedCube.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.TruncatedCube(6.74*nm, 2*nm)
diff --git a/rawEx/ff/TruncatedSphere.py b/rawEx/ff/TruncatedSphere.py
index 215e2b8acbc..8b49deb8cc0 100755
--- a/rawEx/ff/TruncatedSphere.py
+++ b/rawEx/ff/TruncatedSphere.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.TruncatedSphere(7*nm, 4.25*nm, 1*nm)
diff --git a/rawEx/ff/TruncatedSpheroid.py b/rawEx/ff/TruncatedSpheroid.py
index 8cf1059ddfb..540dd512359 100755
--- a/rawEx/ff/TruncatedSpheroid.py
+++ b/rawEx/ff/TruncatedSpheroid.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, std_simulations
+import matplotlib.pyplot as plt
 
 material_particle = ba.RefractiveMaterial("Particle", 1e-6, 0)
 ff = ba.TruncatedSpheroid(7*nm, 6.*nm, 1.9, 2.3*nm)
diff --git a/rawEx/offspec/Offspec1.py b/rawEx/offspec/Offspec1.py
index 3e43ae7b1fb..6049db86e92 100755
--- a/rawEx/offspec/Offspec1.py
+++ b/rawEx/offspec/Offspec1.py
@@ -5,6 +5,7 @@ Sample is a grating on a substrate, modelled by a 1d lattice of long boxes
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/offspec/OffspecLambda.py b/rawEx/offspec/OffspecLambda.py
index 5ff9bacd8c1..3f002ea1a70 100755
--- a/rawEx/offspec/OffspecLambda.py
+++ b/rawEx/offspec/OffspecLambda.py
@@ -5,6 +5,7 @@ Sample is a grating on a substrate, modelled by a 1d lattice of long boxes
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/offspec/OffspecResolved.py b/rawEx/offspec/OffspecResolved.py
index ace0506de8f..271ff5dfbd8 100755
--- a/rawEx/offspec/OffspecResolved.py
+++ b/rawEx/offspec/OffspecResolved.py
@@ -4,6 +4,7 @@ Same sample as in Offspec1, but simulated as GISAS, i.e. without integration in
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/ApproximationDA.py b/rawEx/scatter2d/ApproximationDA.py
index 63b997348f0..599b7385a9c 100755
--- a/rawEx/scatter2d/ApproximationDA.py
+++ b/rawEx/scatter2d/ApproximationDA.py
@@ -4,6 +4,7 @@ Cylinders of two different sizes in Decoupling Approximation
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/ApproximationLMA.py b/rawEx/scatter2d/ApproximationLMA.py
index 357d17f242a..dc0ae808ae7 100755
--- a/rawEx/scatter2d/ApproximationLMA.py
+++ b/rawEx/scatter2d/ApproximationLMA.py
@@ -4,6 +4,7 @@ Cylinders of two different sizes in Local Monodisperse Approximation
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/ApproximationSSCA.py b/rawEx/scatter2d/ApproximationSSCA.py
index 883bb8768d7..746cb894079 100755
--- a/rawEx/scatter2d/ApproximationSSCA.py
+++ b/rawEx/scatter2d/ApproximationSSCA.py
@@ -4,7 +4,7 @@ Cylinders of two different sizes in Size-Spacing Coupling Approximation
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
-
+import matplotlib.pyplot as plt
 
 def get_sample():
     """
diff --git a/rawEx/scatter2d/BeamDivergence.py b/rawEx/scatter2d/BeamDivergence.py
index 69794f0cd67..249c9ee5a25 100755
--- a/rawEx/scatter2d/BeamDivergence.py
+++ b/rawEx/scatter2d/BeamDivergence.py
@@ -4,6 +4,7 @@ Cylinder form factor in DWBA with beam divergence
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/BiMaterialCylinders.py b/rawEx/scatter2d/BiMaterialCylinders.py
index 43e49d21537..875e34e0290 100755
--- a/rawEx/scatter2d/BiMaterialCylinders.py
+++ b/rawEx/scatter2d/BiMaterialCylinders.py
@@ -5,6 +5,7 @@ Particle crosses air/substrate interface.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/BoxesWithSpecularPeak.py b/rawEx/scatter2d/BoxesWithSpecularPeak.py
index 14fc6846324..5d939557ea0 100755
--- a/rawEx/scatter2d/BoxesWithSpecularPeak.py
+++ b/rawEx/scatter2d/BoxesWithSpecularPeak.py
@@ -4,6 +4,7 @@ GISAS simulation that includes the specular peak
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/BroadSpecular.py b/rawEx/scatter2d/BroadSpecular.py
index 8c052ef68f9..3bd78caf70c 100755
--- a/rawEx/scatter2d/BroadSpecular.py
+++ b/rawEx/scatter2d/BroadSpecular.py
@@ -1,6 +1,8 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample():
     # Define materials
diff --git a/rawEx/scatter2d/ConstantBackground.py b/rawEx/scatter2d/ConstantBackground.py
index a3b0ffbce21..d266bf59354 100755
--- a/rawEx/scatter2d/ConstantBackground.py
+++ b/rawEx/scatter2d/ConstantBackground.py
@@ -4,6 +4,7 @@ Cylinder form factor in DWBA with constant background
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/CoreShellNanoparticles.py b/rawEx/scatter2d/CoreShellNanoparticles.py
index 61b2e95e1c5..87eaed9ae20 100755
--- a/rawEx/scatter2d/CoreShellNanoparticles.py
+++ b/rawEx/scatter2d/CoreShellNanoparticles.py
@@ -4,6 +4,7 @@ Core shell nanoparticles
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/CoreShellNanoparticles2.py b/rawEx/scatter2d/CoreShellNanoparticles2.py
index 7cd17f23831..d5e266a36a9 100755
--- a/rawEx/scatter2d/CoreShellNanoparticles2.py
+++ b/rawEx/scatter2d/CoreShellNanoparticles2.py
@@ -5,6 +5,7 @@ Alternative implementation using particle composition
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, std_samples, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/CorrelatedRoughness.py b/rawEx/scatter2d/CorrelatedRoughness.py
index 484442d0996..a77ecc2542b 100755
--- a/rawEx/scatter2d/CorrelatedRoughness.py
+++ b/rawEx/scatter2d/CorrelatedRoughness.py
@@ -4,6 +4,7 @@ MultiLayer with correlated roughness
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/CosineRipplesAtRectLattice.py b/rawEx/scatter2d/CosineRipplesAtRectLattice.py
index 799942dc012..cac991481cb 100755
--- a/rawEx/scatter2d/CosineRipplesAtRectLattice.py
+++ b/rawEx/scatter2d/CosineRipplesAtRectLattice.py
@@ -4,6 +4,7 @@ Cosine ripple on a 2D lattice
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/CustomFormFactor.py b/rawEx/scatter2d/CustomFormFactor.py
index 222722a6254..6e29a1ab0f0 100755
--- a/rawEx/scatter2d/CustomFormFactor.py
+++ b/rawEx/scatter2d/CustomFormFactor.py
@@ -5,6 +5,7 @@ Custom form factor in DWBA.
 import cmath
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, angstrom, nm
+import matplotlib.pyplot as plt
 
 
 def sinc(x):
diff --git a/rawEx/scatter2d/Cylinders.py b/rawEx/scatter2d/Cylinders.py
index ddb59a33e28..eb305ac4021 100755
--- a/rawEx/scatter2d/Cylinders.py
+++ b/rawEx/scatter2d/Cylinders.py
@@ -4,6 +4,7 @@ Basic example of a DWBA simulation of a GISAS experiment.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/CylindersAndPrisms.py b/rawEx/scatter2d/CylindersAndPrisms.py
index 6d285cca24e..6f1cc68efa7 100755
--- a/rawEx/scatter2d/CylindersAndPrisms.py
+++ b/rawEx/scatter2d/CylindersAndPrisms.py
@@ -4,6 +4,8 @@ Mixture of cylinders and prisms without interference
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample():
     """
diff --git a/rawEx/scatter2d/CylindersInAverageLayer.py b/rawEx/scatter2d/CylindersInAverageLayer.py
index 1379daa0e5c..46099607e26 100755
--- a/rawEx/scatter2d/CylindersInAverageLayer.py
+++ b/rawEx/scatter2d/CylindersInAverageLayer.py
@@ -4,6 +4,7 @@ Square lattice of cylinders inside finite layer with usage of average material
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample(cyl_height=5*nm):
diff --git a/rawEx/scatter2d/CylindersInBA.py b/rawEx/scatter2d/CylindersInBA.py
index 4e0b7504a5a..6b01d988a75 100755
--- a/rawEx/scatter2d/CylindersInBA.py
+++ b/rawEx/scatter2d/CylindersInBA.py
@@ -4,6 +4,7 @@ Cylinder form factor in Born approximation
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/DodecahedraSAS.py b/rawEx/scatter2d/DodecahedraSAS.py
index ed90417e31f..714ad15cac7 100755
--- a/rawEx/scatter2d/DodecahedraSAS.py
+++ b/rawEx/scatter2d/DodecahedraSAS.py
@@ -5,6 +5,7 @@ Sample is a dilute assembly of ordered dodecahedra.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/HalfSpheresInAverageTopLayer.py b/rawEx/scatter2d/HalfSpheresInAverageTopLayer.py
index e067c325504..0107d9cafbc 100755
--- a/rawEx/scatter2d/HalfSpheresInAverageTopLayer.py
+++ b/rawEx/scatter2d/HalfSpheresInAverageTopLayer.py
@@ -5,6 +5,7 @@ and slicing
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/HexagonalLatticesWithBasis.py b/rawEx/scatter2d/HexagonalLatticesWithBasis.py
index e35531c5793..70196f591a4 100755
--- a/rawEx/scatter2d/HexagonalLatticesWithBasis.py
+++ b/rawEx/scatter2d/HexagonalLatticesWithBasis.py
@@ -4,6 +4,7 @@ Spheres on two hexagonal close packed layers
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Interference1DLattice.py b/rawEx/scatter2d/Interference1DLattice.py
index d7fe1f55daa..a8d3e98339a 100755
--- a/rawEx/scatter2d/Interference1DLattice.py
+++ b/rawEx/scatter2d/Interference1DLattice.py
@@ -6,6 +6,7 @@ large-particle form factor oscillations.
 """
 import bornagain as ba
 from bornagain import angstrom, ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Interference1DRadialParacrystal.py b/rawEx/scatter2d/Interference1DRadialParacrystal.py
index bffc097f492..c0036ecb4f6 100755
--- a/rawEx/scatter2d/Interference1DRadialParacrystal.py
+++ b/rawEx/scatter2d/Interference1DRadialParacrystal.py
@@ -4,6 +4,7 @@ radial paracrystal
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Interference2DCenteredSquareLattice.py b/rawEx/scatter2d/Interference2DCenteredSquareLattice.py
index 68d47046316..7025e309600 100755
--- a/rawEx/scatter2d/Interference2DCenteredSquareLattice.py
+++ b/rawEx/scatter2d/Interference2DCenteredSquareLattice.py
@@ -4,6 +4,8 @@
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
+
 
 r = 3*nm  # particle radius
 a = 25*nm  # lattice constant
diff --git a/rawEx/scatter2d/Interference2DParacrystal.py b/rawEx/scatter2d/Interference2DParacrystal.py
index 8021f66ed1f..81f51b6f8f6 100755
--- a/rawEx/scatter2d/Interference2DParacrystal.py
+++ b/rawEx/scatter2d/Interference2DParacrystal.py
@@ -4,6 +4,7 @@
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Interference2DRotatedSquareLattice.py b/rawEx/scatter2d/Interference2DRotatedSquareLattice.py
index 22f0aae3a48..6d453a9e02c 100755
--- a/rawEx/scatter2d/Interference2DRotatedSquareLattice.py
+++ b/rawEx/scatter2d/Interference2DRotatedSquareLattice.py
@@ -4,6 +4,7 @@ Cylinders on a rotated 2D lattice
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Interference2DSquareFiniteLattice.py b/rawEx/scatter2d/Interference2DSquareFiniteLattice.py
index 13658d1ccd8..5d74ca24998 100755
--- a/rawEx/scatter2d/Interference2DSquareFiniteLattice.py
+++ b/rawEx/scatter2d/Interference2DSquareFiniteLattice.py
@@ -4,6 +4,7 @@ Cylinders on a 2D square lattice
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, nm2
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/LatticeOrientationDistribution.py b/rawEx/scatter2d/LatticeOrientationDistribution.py
index 5d20da129a8..fbdec4dbdb6 100755
--- a/rawEx/scatter2d/LatticeOrientationDistribution.py
+++ b/rawEx/scatter2d/LatticeOrientationDistribution.py
@@ -5,6 +5,7 @@ square lattices of cylinders on a substrate.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/MagneticCylinders1.py b/rawEx/scatter2d/MagneticCylinders1.py
index f5cd1c4ee93..296a24386b6 100755
--- a/rawEx/scatter2d/MagneticCylinders1.py
+++ b/rawEx/scatter2d/MagneticCylinders1.py
@@ -4,6 +4,7 @@ Functional test: Magnetic cylinders in DWBA with zero magnetic field
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/MagneticSpheres.py b/rawEx/scatter2d/MagneticSpheres.py
index 11cc9970406..73af5e23295 100755
--- a/rawEx/scatter2d/MagneticSpheres.py
+++ b/rawEx/scatter2d/MagneticSpheres.py
@@ -4,6 +4,7 @@ Spin-flip scattering by magnetic spheres
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Mesocrystal.py b/rawEx/scatter2d/Mesocrystal.py
index adeff4d4467..f18b7538e4d 100755
--- a/rawEx/scatter2d/Mesocrystal.py
+++ b/rawEx/scatter2d/Mesocrystal.py
@@ -4,6 +4,7 @@ Cylindrical mesocrystal on a substrate
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Mesocrystal3.py b/rawEx/scatter2d/Mesocrystal3.py
index 7fd326730fe..4c93f2ad6bc 100755
--- a/rawEx/scatter2d/Mesocrystal3.py
+++ b/rawEx/scatter2d/Mesocrystal3.py
@@ -4,6 +4,7 @@ Cylindrical mesocrystal with hexagonal lattice of soft spheres on a substrate
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/Mesocrystal4.py b/rawEx/scatter2d/Mesocrystal4.py
index e22b5ce6f8d..dc78309a4f0 100755
--- a/rawEx/scatter2d/Mesocrystal4.py
+++ b/rawEx/scatter2d/Mesocrystal4.py
@@ -6,6 +6,7 @@ With slicing.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/ObsoleteSimN.py b/rawEx/scatter2d/ObsoleteSimN.py
index d6bc5c6679a..514aad3d58f 100755
--- a/rawEx/scatter2d/ObsoleteSimN.py
+++ b/rawEx/scatter2d/ObsoleteSimN.py
@@ -4,6 +4,7 @@ Basic example of a DWBA simulation of a GISAS experiment.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/ParticleAcrossInterface.py b/rawEx/scatter2d/ParticleAcrossInterface.py
index 29d9155f00c..faeef4d0fc1 100755
--- a/rawEx/scatter2d/ParticleAcrossInterface.py
+++ b/rawEx/scatter2d/ParticleAcrossInterface.py
@@ -4,6 +4,7 @@ Scattering from particles that cross the interface between vacuum and substrate.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/PolarizedSANS.py b/rawEx/scatter2d/PolarizedSANS.py
index b5ed5c16af5..52d41a2131a 100755
--- a/rawEx/scatter2d/PolarizedSANS.py
+++ b/rawEx/scatter2d/PolarizedSANS.py
@@ -5,6 +5,7 @@ simulated with BornAgain.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm, R3
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/PolydisperseCylinders.py b/rawEx/scatter2d/PolydisperseCylinders.py
index 06b9580f3f4..7f4af56cfea 100755
--- a/rawEx/scatter2d/PolydisperseCylinders.py
+++ b/rawEx/scatter2d/PolydisperseCylinders.py
@@ -4,6 +4,8 @@ GISAS by cylinders with Gaussian radius distribution on a substrate.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample():
     material_particle = ba.RefractiveMaterial("Particle", 0.0006, 2e-08)
diff --git a/rawEx/scatter2d/RectangularGrating.py b/rawEx/scatter2d/RectangularGrating.py
index 27b9ee472c3..66d771a67e0 100755
--- a/rawEx/scatter2d/RectangularGrating.py
+++ b/rawEx/scatter2d/RectangularGrating.py
@@ -6,6 +6,7 @@ large-particle form factor oscillations.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, micrometer, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample(lattice_rotation_angle=0*deg):
diff --git a/rawEx/scatter2d/Resolution.py b/rawEx/scatter2d/Resolution.py
index 1dc5ac03bf8..3faed455728 100755
--- a/rawEx/scatter2d/Resolution.py
+++ b/rawEx/scatter2d/Resolution.py
@@ -4,6 +4,7 @@ Basic GISAS from oriented boxes, with different detector resolutions.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/RotatedPyramids.py b/rawEx/scatter2d/RotatedPyramids.py
index e9cd59861ce..6c01b921530 100755
--- a/rawEx/scatter2d/RotatedPyramids.py
+++ b/rawEx/scatter2d/RotatedPyramids.py
@@ -4,6 +4,7 @@ Rotated pyramids on top of substrate
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/RoughAndSpecular.py b/rawEx/scatter2d/RoughAndSpecular.py
index 68d85a2696d..475eb41d6e7 100755
--- a/rawEx/scatter2d/RoughAndSpecular.py
+++ b/rawEx/scatter2d/RoughAndSpecular.py
@@ -1,6 +1,8 @@
 #!/usr/bin/env python3
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample():
     # Define materials
@@ -12,7 +14,7 @@ def get_sample():
     # Define roughness
     autocorr_1 = ba.K_CorrelationModel(1.1, 0.3, 5*nm)
     autocorr_2 = ba.K_CorrelationModel(2.3, 0.3, 5*nm)
-    
+
     interlayer = ba.TanhInterlayer()
 
     roughness_1 = ba.LayerRoughness(autocorr_1, interlayer)
diff --git a/rawEx/scatter2d/SlicedLayer.py b/rawEx/scatter2d/SlicedLayer.py
index 62084964d52..572188019b5 100755
--- a/rawEx/scatter2d/SlicedLayer.py
+++ b/rawEx/scatter2d/SlicedLayer.py
@@ -4,6 +4,7 @@ Spheres in a finite layer. For testing the slicing machinery.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/SpheresAtHexLattice.py b/rawEx/scatter2d/SpheresAtHexLattice.py
index 6032decc27d..1eb14f21fb9 100755
--- a/rawEx/scatter2d/SpheresAtHexLattice.py
+++ b/rawEx/scatter2d/SpheresAtHexLattice.py
@@ -4,6 +4,7 @@ Spheres on a hexagonal lattice
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/scatter2d/TriangularRipple.py b/rawEx/scatter2d/TriangularRipple.py
index 86212f690bc..db447049e43 100755
--- a/rawEx/scatter2d/TriangularRipple.py
+++ b/rawEx/scatter2d/TriangularRipple.py
@@ -4,6 +4,7 @@
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/specular/AlternatingLayers1.py b/rawEx/specular/AlternatingLayers1.py
index 79ba1281745..35351fa263e 100755
--- a/rawEx/specular/AlternatingLayers1.py
+++ b/rawEx/specular/AlternatingLayers1.py
@@ -5,6 +5,7 @@ The sample consists of 20 alternating Ti and Ni layers.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, angstrom
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/specular/AlternatingLayers2.py b/rawEx/specular/AlternatingLayers2.py
index 25fe94ab49a..0b2e45cb112 100755
--- a/rawEx/specular/AlternatingLayers2.py
+++ b/rawEx/specular/AlternatingLayers2.py
@@ -5,6 +5,7 @@ using a standard sample from Wrap/Python/std_samples.py.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, angstrom, std_samples
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/specular/Basic.py b/rawEx/specular/Basic.py
index 76715296d8b..17873962b37 100755
--- a/rawEx/specular/Basic.py
+++ b/rawEx/specular/Basic.py
@@ -4,6 +4,8 @@
 import numpy as np
 import bornagain as ba
 from bornagain import ba_plot as bp, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample():
     # Materials
diff --git a/rawEx/specular/BeamFullDivergence.py b/rawEx/specular/BeamFullDivergence.py
index 11dd732f978..3658940d660 100755
--- a/rawEx/specular/BeamFullDivergence.py
+++ b/rawEx/specular/BeamFullDivergence.py
@@ -5,6 +5,7 @@ divergence in reflectometry calculations with BornAgain.
 """
 import bornagain as ba
 from bornagain import angstrom, ba_plot as bp, deg, std_samples
+import matplotlib.pyplot as plt
 
 # input parameters
 wavelength = 1.54*angstrom
diff --git a/rawEx/specular/Distributions.py b/rawEx/specular/Distributions.py
index 80aa9e75a82..08d0f93a4d9 100755
--- a/rawEx/specular/Distributions.py
+++ b/rawEx/specular/Distributions.py
@@ -4,6 +4,8 @@ Reflectivity of a multilayer with different wavelength distributions.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, std_samples
+import matplotlib.pyplot as plt
+
 
 lambda0 = 0.154
 
diff --git a/rawEx/specular/MagneticLayer.py b/rawEx/specular/MagneticLayer.py
index 0bec3b2a318..a389b70d2db 100755
--- a/rawEx/specular/MagneticLayer.py
+++ b/rawEx/specular/MagneticLayer.py
@@ -6,6 +6,7 @@ Sample is a magnetic layer.
 import bornagain as ba
 from bornagain import angstrom, ba_plot as bp, deg, nm, R3
 from math import sin, cos
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/specular/SpecularSimulationWithRoughness.py b/rawEx/specular/SpecularSimulationWithRoughness.py
index e876f424391..90ee2b0c682 100755
--- a/rawEx/specular/SpecularSimulationWithRoughness.py
+++ b/rawEx/specular/SpecularSimulationWithRoughness.py
@@ -6,6 +6,7 @@ with a rough sample using BornAgain.
 """
 import bornagain as ba
 from bornagain import angstrom, ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/specular/TOFRWithResolution.py b/rawEx/specular/TOFRWithResolution.py
index 51bfcf7e692..de286ecf5e4 100755
--- a/rawEx/specular/TOFRWithResolution.py
+++ b/rawEx/specular/TOFRWithResolution.py
@@ -13,6 +13,7 @@ materials only.
 import numpy as np
 import bornagain as ba
 from bornagain import ba_plot as bp, std_samples
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/specular/TimeOfFlightReflectometry.py b/rawEx/specular/TimeOfFlightReflectometry.py
index 396fe96188e..3fbf6fc1f3e 100755
--- a/rawEx/specular/TimeOfFlightReflectometry.py
+++ b/rawEx/specular/TimeOfFlightReflectometry.py
@@ -11,6 +11,7 @@ materials only.
 import numpy as np
 import bornagain as ba
 from bornagain import ba_plot as bp, std_samples
+import matplotlib.pyplot as plt
 
 
 def get_sample():
diff --git a/rawEx/varia/Depthprobe1.py b/rawEx/varia/Depthprobe1.py
index 0cced05a908..20d0dc4d421 100755
--- a/rawEx/varia/Depthprobe1.py
+++ b/rawEx/varia/Depthprobe1.py
@@ -5,7 +5,7 @@ Computes intensity as function of incident angle alpha and depth z.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
-
+import matplotlib.pyplot as plt
 
 def get_sample():
     # Materials
diff --git a/rawEx/varia/OpaqueProfile1.py b/rawEx/varia/OpaqueProfile1.py
index 285ec5212c2..fe6a727f141 100755
--- a/rawEx/varia/OpaqueProfile1.py
+++ b/rawEx/varia/OpaqueProfile1.py
@@ -4,6 +4,7 @@ Depth profile of an opaque sample.
 """
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
 
 #  beam data
 ai_min = 0  # minimum incident angle
diff --git a/rawEx/varia/Resonator.py b/rawEx/varia/Resonator.py
index 68dd785ba11..68abe670ff0 100755
--- a/rawEx/varia/Resonator.py
+++ b/rawEx/varia/Resonator.py
@@ -9,6 +9,8 @@ The z axis points from D2O to Si; z=0 is at the Si/Ti interface.
 """
 import bornagain as ba
 from bornagain import angstrom, ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 # layer thicknesses in angstroms
 t_Ti = 130*angstrom
diff --git a/rawEx/varia/TransmissionVsAlpha.py b/rawEx/varia/TransmissionVsAlpha.py
index f15ce7bf175..a06875b5daa 100755
--- a/rawEx/varia/TransmissionVsAlpha.py
+++ b/rawEx/varia/TransmissionVsAlpha.py
@@ -6,6 +6,8 @@ Intensity as function of incident angle alpha_i for a few depths z.
 
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample(depth):
     material_vac = ba.RefractiveMaterial("Vacuum", 0, 0)
diff --git a/rawEx/varia/TransmissionVsDepth.py b/rawEx/varia/TransmissionVsDepth.py
index 0cfd119f204..a3cf391df0b 100755
--- a/rawEx/varia/TransmissionVsDepth.py
+++ b/rawEx/varia/TransmissionVsDepth.py
@@ -6,6 +6,8 @@ Intensity as function of depth z for a few incident angles alpha_i.
 
 import bornagain as ba
 from bornagain import ba_plot as bp, deg, nm
+import matplotlib.pyplot as plt
+
 
 def get_sample():
     material_vac = ba.RefractiveMaterial("Vacuum", 0, 0)
-- 
GitLab