Update after NumPy 2.0+ removal of ndarray.newbyteorder

MathEXLab · Nov 18, 2024 · 277d5de · 277d5de
1 parent 51a12d9
commit 277d5de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyspod/utils/parallel.py b/pyspod/utils/parallel.py
@@ -161,7 +161,7 @@ def _blockdist(N, size, rank):
 def allreduce(data, comm):
     if comm is not None:
         MPI = _get_module_MPI(comm)
-        data = data.newbyteorder('=')
+        data = data.view(data.dtype.newbyteorder('='))
         data_reduced = np.zeros_like(data)
         comm.Barrier()
         comm.Allreduce(data, data_reduced, op=MPI.SUM)