Fix remote bug.

h2oai · Jun 11, 2015 · fdebe7a · fdebe7a
1 parent 2e67901
commit fdebe7a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/h2o-algos/src/test/java/hex/deeplearning/DeepLearningTest.java b/h2o-algos/src/test/java/hex/deeplearning/DeepLearningTest.java
@@ -534,6 +534,7 @@ public void basicDL(String fnametrain, String hexnametrain, String fnametest, Pr
         Vec resp = frTrain.lastVec().toEnum();
         frTrain.remove(frTrain.vecs().length - 1).remove();
         frTrain.add("Response", resp);
+        DKV.put(frTrain);
         dl._train = frTrain._key;
         dl._response_column = ((Frame) DKV.getGet(dl._train)).lastVecName();
         dl._export_weights_and_biases = true;