Merge branch 'topic/default/fix-check' into 'branch/default'

Fix import check + prepare 0.4.2

See merge request fluiddyn/fluidfft!69

Author: paugier

CHANGES.md

@@ -2,6 +2,11 @@
 
 See also the [unreleased changes].
 
+## [0.4.2] (2024-08-22)
+
+- Compatibility mpi4py 4.0
+- Fix check import classes and `FLUIDFFT_DISABLE_IMPORT_CHECK`.
+
 ## [0.4.1] (2024-07-24)
 
 - Support for Numpy 2.0 and Python 3.12
@@ -114,4 +119,5 @@ See also the [unreleased changes].
 
 [0.4.0]: https://foss.heptapod.net/fluiddyn/fluidfft/-/compare/0.3.5...0.4.0
 [0.4.1]: https://foss.heptapod.net/fluiddyn/fluidfft/-/compare/0.4.0...0.4.1
-[unreleased changes]: https://foss.heptapod.net/fluiddyn/fluidfft/-/compare/0.4.1...branch%2Fdefault
+[0.4.2]: https://foss.heptapod.net/fluiddyn/fluidfft/-/compare/0.4.1...0.4.2
+[unreleased changes]: https://foss.heptapod.net/fluiddyn/fluidfft/-/compare/0.4.2...branch%2Fdefault

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = 'mesonpy'
 
 [project]
 name = "fluidfft"
-version = "0.4.1"
+version = "0.4.2"
 description = "Efficient and easy Fast Fourier Transform (FFT) for Python."
 authors = [
     {name = "Pierre Augier", email = "[email protected]"},

src/fluidfft/__init__.py

@@ -177,6 +177,9 @@ def _check_failure(method):
     if not any(method.endswith(postfix) for postfix in ("pfft", "p3dfft")):
         return False
 
+    if os.environ.get("FLUIDFFT_DISABLE_IMPORT_CHECK"):
+        return False
+
     # for few methods, try before real import because importing can lead to
     # a fatal error (Illegal instruction)
     if mpi.rank == 0:
@@ -186,7 +189,11 @@ def _check_failure(method):
             env = {
                 key: value
                 for key, value in os.environ.items()
-                if not ("MPI" in key or key.startswith("PMI_"))
+                if not (
+                    "MPI" in key
+                    or key.startswith("PMI_")
+                    or key.startswith("PMIX_")
+                )
             }
         else:
             env = os.environ