Sampling no longer fails on empty compartments.

tgvaughan · Jun 22, 2016 · ed923fc · ed923fc
1 parent 8e1b8b3
commit ed923fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/master/postprocessors/LineageSampler.java b/src/master/postprocessors/LineageSampler.java
@@ -304,7 +304,7 @@ private static void collectNodes(Node node, double samplingTime,
      */
     private static List<Node> getSampledLineages(List<Node> nodesToSample, int nSamples) {
         List<Node> sampledNodes = Lists.newArrayList();
-        while (nSamples > 0 && !nodesToSample.isEmpty()) {
+        while (nSamples > 0 && nodesToSample != null && !nodesToSample.isEmpty()) {
             int index = Randomizer.nextInt(nodesToSample.size());
             Node sampledNode = nodesToSample.remove(index);
             sampledNodes.add(sampledNode);