Refact: suppress external logger info messages.

OHBA-analysis · May 21, 2024 · 50661c1 · 50661c1
1 parent 82a2856
commit 50661c1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/osl_dynamics/analysis/power.py b/osl_dynamics/analysis/power.py
@@ -247,6 +247,9 @@ def parcel_vector_to_voxel_grid(mask_file, parcellation_file, vector):
         Value at each voxel. Shape is (x, y, z), where :code:`x`,
         :code:`y` and :code:`z` correspond to 3D voxel locations.
     """
+    # Suppress INFO messages from nibabel
+    logging.getLogger("nibabel.global").setLevel(logging.ERROR)
+
     # Validation
     mask_file = files.check_exists(mask_file, files.mask.directory)
     parcellation_file = files.check_exists(

diff --git a/osl_dynamics/models/hmm.py b/osl_dynamics/models/hmm.py
@@ -217,6 +217,9 @@ def fit(self, dataset, epochs=None, use_tqdm=False, verbose=1, **kwargs):
             Dictionary with history of the loss and learning rates (:code:`lr`
             and :code:`rho`).
         """
+        # Suppress INFO messages from numba
+        logging.getLogger("numba.core.transforms").setLevel(logging.ERROR)
+
         if epochs is None:
             epochs = self.config.n_epochs