Cast remain_dims to int

Author: kburns

dedalus/core/arithmetic.py

@@ -878,7 +878,7 @@ def __init__(self, domain, layout, broadcast_dims):
         deploy_dims = deploy_dims_ext[~layout.local]
         # Build subcomm or skip casting
         if any(deploy_dims):
-            self.subcomm = domain.dist.comm_cart.Sub(remain_dims=deploy_dims)
+            self.subcomm = domain.dist.comm_cart.Sub(remain_dims=deploy_dims.astype(int))
         else:
             self.cast = self._skip_cast
 

dedalus/core/field.py

@@ -874,7 +874,7 @@ def broadcast_ghosts(self, output_nonconst_dims):
         if not any(deploy_dims):
             return self.data
         # Broadcast on subgrid communicator
-        comm_sub = self.domain.dist.comm_cart.Sub(remain_dims=deploy_dims)
+        comm_sub = self.domain.dist.comm_cart.Sub(remain_dims=deploy_dims.astype(int))
         data = None
         if comm_sub.rank == 0:
             data = self.data