All fixed with trimmed testcase

Mridul Jain · Mridul Jain · commit 9147590b6ec4 · 2016-11-04T10:15:21.000-07:00
diff --git a/caffe-grid/src/test/scala/com/yahoo/ml/caffe/tools/ToolTest.scala b/caffe-grid/src/test/scala/com/yahoo/ml/caffe/tools/ToolTest.scala
@@ -96,8 +96,8 @@ class ToolTest extends FunSuite with BeforeAndAfterAll {
 
     FileUtils.deleteQuietly(new File(cocoImageCaptionDF))
     val df_image_caption = Conversions.Coco2ImageCaptionFile(sqlContext, cocoJson, 4)
-    val rdd_input_captions = inputDF2PairRDD(df_image_caption)
-    val count = rdd_input_captions.count.toInt
+//    val rdd_input_captions = inputDF2PairRDD(df_image_caption)
+    val count = df_image_caption.count.toInt
 
     FileUtils.deleteQuietly(new File(cocoEmbeddingDF))
     FileUtils.deleteQuietly(new File(cocoVocab))
