fix: pacify mypy

Author: oesteban

scripts/compute_nufo.py

@@ -164,7 +164,7 @@ def main() -> None:
         _, brain_mask = median_otsu(dwi_data, vol_idx=[0])
 
     dwi_data_masked = dwi_data.copy()
-    dwi_data_masked[~brain_mask, :] = 0
+    dwi_data_masked[~brain_mask.astype(bool), :] = 0
 
     # Create a CSD model
     response, ratio = auto_response_ssst(

src/nifreeze/model/_dipy.py

@@ -24,8 +24,6 @@
 
 from __future__ import annotations
 
-from typing import Any
-
 import numpy as np
 from dipy.core.gradients import GradientTable
 from dipy.reconst.base import ReconstModel
@@ -139,7 +137,7 @@ def fit(
         self,
         data: np.ndarray,
         gtab: GradientTable | np.ndarray,
-        mask: np.ndarray[bool, Any] | None = None,
+        mask: np.ndarray | None = None,
         random_state: int = 0,
     ) -> GPFit:
         """Fit method of the DTI model class