QuasarCode 0.22.0__tar.gz → 0.22.2__tar.gz

{quasarcode-0.22.0 → quasarcode-0.22.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: QuasarCode
-Version: 0.22.0
+Version: 0.22.2
 Summary: A general purpose library for Python applications.
 Project-URL: Documentation, https://github.com/QuasarX1/QuasarCode/tree/dev-python/QuasarCode_Python/README.md
 Project-URL: Issues, https://github.com/QuasarX1/QuasarCode/issues

{quasarcode-0.22.0 → quasarcode-0.22.2}/src/QuasarCode/MPI/_dependant_mpi/_convinience_methods.py

@@ -130,7 +130,7 @@ def mpi_slice(data: Sequence[T], comm: MPI.Intracomm|None = None, rank: int|None
 
 
 
-def mpi_gather_array(data: np.ndarray, comm: MPI.Intracomm|None = None, root: int|None = None, target_buffer: np.ndarray|None = None) -> np.ndarray:
+def mpi_gather_array(data: np.ndarray, comm: MPI.Intracomm|None = None, root: int|None = None, target_buffer: np.ndarray|None = None, return_chunk_sizes: bool = False) -> np.ndarray|tuple[np.ndarray, np.ndarray]:
     """
     Gather numpy array data to the root rank.
     If the resulting array is larger than the MPI buffer length, data will be transmitted point-to-point in rank order instead of using the Gatherv method.
@@ -195,7 +195,10 @@ def mpi_gather_array(data: np.ndarray, comm: MPI.Intracomm|None = None, root: in
                 comm.send(data, dest = root)
             comm.barrier()
 
-    return target_buffer
+    if return_chunk_sizes:
+        return target_buffer, input_buffer_lengths_first_dimension
+    else:
+        return target_buffer
 
 
 
@@ -212,7 +215,7 @@ def mpi_scatter_array(data: np.ndarray|None, elements_this_rank: int|None = None
     comm.barrier()
 
     if comm.rank == root:
-        can_continue = data is None
+        can_continue = data is not None
     synchronyse("can_continue", root = root, comm = comm)
     if not can_continue:
         raise TypeError("Root rank did not provide any data.")
@@ -268,8 +271,8 @@ def mpi_scatter_array(data: np.ndarray|None, elements_this_rank: int|None = None
     synchronyse("local_buffer_shape_after_first_dimension", root = root, comm = comm)
 
     if target_buffer is not None:
-        if any(comm.allgather(target_buffer is not None and (target_buffer.shape[0] != elements_this_rank or target_buffer.shape[:1] != local_buffer_shape_after_first_dimension))):
-            raise BufferError("Output bufferes provided to ranks did not match their respective expected sizes/shapes.")
+        if any(comm.allgather(target_buffer is not None and (target_buffer.shape[0] != elements_this_rank or target_buffer.shape[1:] != local_buffer_shape_after_first_dimension))):
+            raise BufferError("Output buffers provided to ranks did not match their respective expected sizes/shapes.")
         comm.barrier()
 
     target_datatype: object

quasarcode-0.22.2/src/QuasarCode/__about__.py

@@ -0,0 +1 @@
+__version__ = "0.22.2"

quasarcode-0.22.0/src/QuasarCode/__about__.py

@@ -1 +0,0 @@
-__version__ = "0.22.0"