Fix errors

scilus · Feb 27, 2024 · 33ce72f · 33ce72f
1 parent 9359b19
commit 33ce72f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/scilpy/image/volume_operations.py b/scilpy/image/volume_operations.py
@@ -14,6 +14,7 @@
 from dipy.segment.mask import crop, median_otsu
 import nibabel as nib
 import numpy as np
+from numpy import ma
 from scipy.ndimage import binary_dilation, gaussian_filter
 
 from scilpy.image.reslice import reslice  # Don't use Dipy's reslice. Buggy.

diff --git a/scilpy/tractanalysis/reproducibility_measures.py b/scilpy/tractanalysis/reproducibility_measures.py
@@ -122,7 +122,7 @@ def approximate_surface_node(roi):
     """
     ind = np.argwhere(roi > 0)
     tree = cKDTree(ind)
-    neighbors = np.sum(7 - tree.query_radius(ind, r=1.0))
+    neighbors = np.sum(7 - tree.query_ball_point(ind, r=1.0))
     count = [len(neighbor) for neighbor in neighbors]
 
     return count

diff --git a/scripts/scil_tractogram_pairwise_comparison.py b/scripts/scil_tractogram_pairwise_comparison.py
@@ -29,14 +29,14 @@
 
 import nibabel as nib
 
+from scilpy.io.streamlines import load_tractogram_with_reference
 from scilpy.io.utils import (add_overwrite_arg, add_reference_arg,
                              assert_inputs_exist,
                              assert_outputs_exist,
                              assert_output_dirs_exist_and_empty,
                              add_processes_arg,
                              add_verbose_arg,
                              is_header_compatible_multiple_files,
-                             load_tractogram_with_reference,
                              validate_nbr_processes)
 from scilpy.tractanalysis.reproducibility_measures import tractogram_pairwise_comparison