svn commit: r1241992 - /mahout/trunk/core/src/main/java/org/apache/mahout/classifier/naivebayes/test/TestNaiveBayesDriver.java

[srowen]

Author: srowen
Date: Wed Feb  8 17:20:09 2012
New Revision: 1241992

URL: http://svn.apache.org/viewvc?rev=1241992&view=rev
Log:
MAHOUT-946 (missed one file)

Modified:
    
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/naivebayes/test/TestNaiveBayesDriver.java

Modified: 
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/naivebayes/test/TestNaiveBayesDriver.java
URL: 
http://svn.apache.org/viewvc/mahout/trunk/core/src/main/java/org/apache/mahout/classifier/naivebayes/test/TestNaiveBayesDriver.java?rev=1241992&r1=1241991&r2=1241992&view=diff
==============================================================================
--- 
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/naivebayes/test/TestNaiveBayesDriver.java
 (original)
+++ 
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/naivebayes/test/TestNaiveBayesDriver.java
 Wed Feb  8 17:20:09 2012
@@ -79,7 +79,10 @@ public class TestNaiveBayesDriver extend
     //testJob.getConfiguration().set(LABEL_KEY, parsedArgs.get("--labels"));
     boolean complementary = parsedArgs.containsKey("--testComplementary");
     testJob.getConfiguration().set(COMPLEMENTARY, 
String.valueOf(complementary));
-    testJob.waitForCompletion(true);
+    boolean succeeded = testJob.waitForCompletion(true);
+    if (!succeeded) {
+      return -1;
+    }
     //load the labels
     Map<Integer, String> labelMap = BayesUtils.readLabelIndex(getConf(), new 
Path(parsedArgs.get("--labelIndex")));