diff --git a/doc/sections/quickstart.rst b/doc/sections/quickstart.rst index f6b6e266d..fde16c2e0 100644 --- a/doc/sections/quickstart.rst +++ b/doc/sections/quickstart.rst @@ -69,7 +69,8 @@ Under ``simdata.f`` you find binning data, in this case a 1d binning plot in the Parallel simulations can invoked from the same launch file for instance by:: - mpirun -n 4 struphy_test.py + pip install -U mpi4py + mpirun -n 4 python struphy_test.py If you want to learn more please check the :ref:`userguide`. diff --git a/src/struphy/models/base.py b/src/struphy/models/base.py index b484397a0..4afcf7709 100644 --- a/src/struphy/models/base.py +++ b/src/struphy/models/base.py @@ -16,7 +16,12 @@ from struphy.feec.basis_projection_ops import BasisProjectionOperators from struphy.feec.mass import WeightedMassOperators from struphy.feec.psydac_derham import SplineFunction -from struphy.fields_background.base import FluidEquilibrium, FluidEquilibriumWithB, MHDequilibrium +from struphy.fields_background.base import ( + FluidEquilibrium, + FluidEquilibriumWithB, + MHDequilibrium, + NumericalMHDequilibrium, +) from struphy.fields_background.equils import HomogenSlab from struphy.fields_background.projected_equils import ( ProjectedFluidEquilibrium, @@ -95,7 +100,7 @@ def setup_domain_and_equil(self, domain: Domain, equil: FluidEquilibrium): """If a numerical equilibirum is used, the domain is taken from this equilibirum.""" if equil is not None: self._equil = equil - if "Numerical" in self.equil.__class__.__name__: + if isinstance(self.equil, NumericalMHDequilibrium): self._domain = self.equil.domain else: self._domain = domain