Add missing NPE checks.

subramaniano · Aug 7, 2014 · e0a5780 · e0a5780
1 parent dd72463
commit e0a5780
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/java/hex/DeepLearningIrisTest.java b/src/test/java/hex/DeepLearningIrisTest.java
@@ -279,7 +279,7 @@ void runFraction(float fraction) {
 //                          }
                                 }
                               } finally {
-                                fpreds.delete();
+                                if (fpreds != null) fpreds.delete();
                               }
                               Log.info("Predicted values: PASS");
 
@@ -317,7 +317,7 @@ void runFraction(float fraction) {
                                   Log.info("Best_model's error : " + bestErr * 100 + "%.");
                                   compareVal(bestErr, best_err, abseps, releps);
                                 } finally {
-                                  bestPredict.delete();
+                                  if (bestPredict != null) bestPredict.delete();
                                 }
                               }