🎨 Rewrite nested if to elsif in VPTree

yoshoku · Apr 12, 2020 · 51080d8 · 51080d8
1 parent a0122d6
commit 51080d8
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/lib/rumale/nearest_neighbors/vp_tree.rb b/lib/rumale/nearest_neighbors/vp_tree.rb
@@ -84,14 +84,12 @@ def search(q, node, k, tau = Float::INFINITY)
           else
             node
           end
+        elsif dist + tau >= node.threshold
+          node.right.n_samples < k ? node : search(q, node.right, k, tau)
+        elsif dist - tau <= node.threshold
+          node.left.n_samples < k ? node : search(q, node.left, k, tau)
         else
-          if dist + tau >= node.threshold
-            node.right.n_samples < k ? node : search(q, node.right, k, tau)
-          elsif dist - tau <= node.threshold
-            node.left.n_samples < k ? node : search(q, node.left, k, tau)
-          else
-            node
-          end
+          node
         end
         # :nocov:
       end