Fix NameError in fix

fluiddyn · Apr 19, 2024 · f87c9bb · f87c9bb
1 parent 42b4f1b
commit f87c9bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/fluidimage/works/piv/fix.py b/src/fluidimage/works/piv/fix.py
@@ -167,7 +167,7 @@ def put_to_nan(inds, explanation):
                 continue
 
             diff_neighbours = np.empty(len(other_peaks_good) + 1)
-            diff_neighbours[0] = sqrt(differences_2[ivec])
+            diff_neighbours[0] = sqrt(differences2[ivec])
 
             for i, (dx, dy, corr) in enumerate(other_peaks_good):
                 diff_neighbours[i + 1] = np.sqrt(

diff --git a/src/fluidimage/works/piv/test_piv.py b/src/fluidimage/works/piv/test_piv.py
@@ -20,8 +20,7 @@ def test_minimal_piv(tmp_path):
     params.piv0.shape_crop_im0 = 32
     params.piv0.grid.overlap = -3
 
-    # still buggy
-    # params.piv0.nb_peaks_to_search = 2
+    params.piv0.nb_peaks_to_search = 2
 
     params.multipass.number = 2