Update Stub Files

examples/aperture/run_aperture.py

@@ -7,6 +7,7 @@
 # -*- coding: utf-8 -*-
 
 import amrex.space3d as amr
+
 from impactx import ImpactX, distribution, elements
 
 # work-around for https://github.com/ECP-WarpX/impactx/issues/499

examples/fodo_tune/run_fodo_tune.py

@@ -7,6 +7,7 @@
 # -*- coding: utf-8 -*-
 
 import amrex.space3d as amr
+
 from impactx import ImpactX, distribution, elements
 
 sim = ImpactX()

examples/initialize_from_array/run_from_array.py

@@ -9,6 +9,7 @@
 import amrex.space3d as amr
 import numpy as np
 import transformation_utilities as pycoord
+
 from impactx import Config, ImpactX, elements
 
 ################

examples/optimize_triplet/run_triplet.py

@@ -7,11 +7,12 @@
 # -*- coding: utf-8 -*-
 
 import amrex.space3d as amr
-import impactx
 import numpy as np
-from impactx import ImpactX, distribution, elements
 from scipy.optimize import minimize
 
+import impactx
+from impactx import ImpactX, distribution, elements
+
 # Call MPI_Init and MPI_Finalize only once:
 if impactx.Config.have_mpi:
     from mpi4py import MPI  # noqa

examples/pytorch_surrogate_model/run_ml_surrogate_15_stage.py

@@ -20,6 +20,8 @@
 
 import numpy as np
 import scipy.optimize as opt
+from surrogate_model_definitions import surrogate_model
+
 from impactx import (
     Config,
     CoordSystem,
@@ -29,7 +31,6 @@
     distribution,
     elements,
 )
-from surrogate_model_definitions import surrogate_model
 
 try:
     import torch

src/python/impactx/__init__.pyi

@@ -17,6 +17,7 @@ from __future__ import annotations
 import os as os
 
 from amrex import space3d as amr
+
 from impactx.impactx_pybind import (
     Config,
     CoordSystem,

tests/python/conftest.py

@@ -3,9 +3,10 @@
 import os
 
 import amrex.space3d as amr
-import impactx
 import pytest
 
+import impactx
+
 if impactx.Config.have_mpi:
     from mpi4py import MPI  # noqa
 

tests/python/test_charge_deposition.py

@@ -11,11 +11,12 @@
 import math
 
 import amrex.space3d as amr
-import impactx
 import matplotlib.pyplot as plt
 import numpy as np
 from conftest import basepath
 
+import impactx
+
 
 def test_charge_deposition(save_png=True):
     """

tests/python/test_dataframe.py

@@ -11,6 +11,7 @@
 
 import matplotlib.pyplot as plt
 import pytest
+
 from impactx import ImpactX, amr, distribution, elements
 
 

tests/python/test_impactx.py

@@ -10,6 +10,7 @@
 import numpy as np
 import pytest
 from conftest import basepath
+
 from impactx import ImpactX, distribution, elements
 
 # FIXME in AMReX via https://github.com/AMReX-Codes/amrex/pull/3727

tests/python/test_transformation.py

@@ -8,6 +8,7 @@
 
 import numpy as np
 import pytest
+
 from impactx import CoordSystem, ImpactX, coordinate_transformation, distribution
 
 

tests/python/test_xopt.py

@@ -10,9 +10,10 @@
 import importlib
 
 import amrex.space3d as amr
-import impactx
 import numpy as np
 import pytest
+
+import impactx
 from impactx import ImpactX, distribution, elements
 
 # configure the test