accept general integral types as MPI ranks

Author: majosm

grudge/discretization.py

@@ -120,11 +120,12 @@ def as_part_id(mesh_part_id):
         # Single-volume
         if mpi_communicator is not None:
             # Accept PartID or rank
+            from numbers import Integral
             def as_part_id(mesh_part_id):
                 if isinstance(mesh_part_id, PartID):
                     return mesh_part_id
-                elif isinstance(mesh_part_id, int):
-                    return PartID(VTAG_ALL, mesh_part_id)
+                elif isinstance(mesh_part_id, Integral):
+                    return PartID(VTAG_ALL, int(mesh_part_id))
                 else:
                     raise TypeError(f"Unable to convert {mesh_part_id} to PartID.")
         else: