parallelized the computation of weighted_* quantities with MPI; updat…

…ed the relevant tests; updated pyproject.toml and setup.py to include the path for mpi4py headers automatically; github test workflow will now run on every push

.github/workflows/tests.yaml

@@ -4,10 +4,8 @@
 # setting up compiler: <https://github.com/marketplace/actions/setup-c-c-compiler>
 
 name: Tests
-on:
-  push:
-    branches:
-      - main
+on: [push]
+
 jobs:
   build:
     runs-on: ${{ matrix.os }}

brahmap/__init__.py

@@ -1,6 +1,15 @@
-from . import interfaces, utilities, linop, mapmakers, _extensions
+import mpi4py
 
-from .utilities import Initialize
+mpi4py.rc.initialize = False
+
+from mpi4py import MPI  # noqa: E402
+
+if MPI.Is_initialized() is False:
+    MPI.Init_thread(required=MPI.THREAD_FUNNELED)
+
+from . import interfaces, utilities, linop, mapmakers, _extensions  # noqa: E402
+
+from .utilities import Initialize, MPI_RAISE_EXCEPTION  # noqa: E402
 
 bMPI = None
 
@@ -11,4 +20,5 @@
     "mapmakers",
     "_extensions",
     "Initialize",
+    "MPI_RAISE_EXCEPTION",
 ]