[mlpack-svn] r15383 - mlpack/trunk/src/mlpack/core/tree/cover_tree

[fastlab-svn at coffeetalk-1.cc.gatech.edu]

Author: rcurtin
Date: Tue Jul  2 16:24:58 2013
New Revision: 15383

Log:
This may fix the segfaulting in the cover tree tests.


Modified:
   mlpack/trunk/src/mlpack/core/tree/cover_tree/dual_tree_traverser_impl.hpp

Modified: mlpack/trunk/src/mlpack/core/tree/cover_tree/dual_tree_traverser_impl.hpp
==============================================================================
--- mlpack/trunk/src/mlpack/core/tree/cover_tree/dual_tree_traverser_impl.hpp	(original)
+++ mlpack/trunk/src/mlpack/core/tree/cover_tree/dual_tree_traverser_impl.hpp	Tue Jul  2 16:24:58 2013
@@ -366,7 +366,11 @@
 
     // If we didn't add anything, then strike this vector from the map.
     if (newScaleVector.size() == 0)
+    {
       referenceMap.erase((*it).first);
+      if (referenceMap.empty())
+        break;
+    }
 
     ++it; // Advance to next scale.
   }