Workaround for numpy 2.0 issue with pyamg (FEniCS#3274)

* Workaround for numpy 2.0 issue * Fix for np.can_cast
schnellerhase · Jun 21, 2024 · 5ebca6c · 5ebca6c
1 parent 58fd478
commit 5ebca6c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/demo/demo_pyamg.py b/python/demo/demo_pyamg.py
@@ -31,7 +31,7 @@
 
 try:
     import pyamg
-except ImportError:
+except (ImportError, AttributeError):
     print('This demo requires pyamg, install using "pip install pyamg"')
     exit(0)
 

diff --git a/python/dolfinx/mesh.py b/python/dolfinx/mesh.py
@@ -496,7 +496,7 @@ def meshtags(
     """
 
     if isinstance(values, int):
-        assert np.can_cast(values, np.int32)
+        assert values >= np.iinfo(np.int32).min and values <= np.iinfo(np.int32).max
         values = np.full(entities.shape, values, dtype=np.int32)
     elif isinstance(values, float):
         values = np.full(entities.shape, values, dtype=np.double)