From 3b8f8d72b31c7488985f408d90ab0305dcbb67c8 Mon Sep 17 00:00:00 2001 From: Hwansung Yu Date: Mon, 3 Sep 2018 18:09:47 +0900 Subject: [PATCH] Refactor unused variable on AnnoyIndexBuildSink class. --- .../s2graph/s2jobs/task/custom/sink/AnnoyIndexBuildSink.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/s2jobs/src/main/scala/org/apache/s2graph/s2jobs/task/custom/sink/AnnoyIndexBuildSink.scala b/s2jobs/src/main/scala/org/apache/s2graph/s2jobs/task/custom/sink/AnnoyIndexBuildSink.scala index c9a590d6..88daa1bb 100644 --- a/s2jobs/src/main/scala/org/apache/s2graph/s2jobs/task/custom/sink/AnnoyIndexBuildSink.scala +++ b/s2jobs/src/main/scala/org/apache/s2graph/s2jobs/task/custom/sink/AnnoyIndexBuildSink.scala @@ -32,9 +32,9 @@ class AnnoyIndexBuildSink(queryName: String, conf: TaskConf) extends Sink(queryN override def write(inputDF: DataFrame): Unit = { val df = repartition(preprocess(inputDF), inputDF.sparkSession.sparkContext.defaultParallelism) - if (inputDF.isStreaming) throw new IllegalStateException("AnnoyIndexBuildSink can not be run as streaming.") + if (df.isStreaming) throw new IllegalStateException("AnnoyIndexBuildSink can not be run as streaming.") else { - ALSModelProcess.buildAnnoyIndex(conf, inputDF) + ALSModelProcess.buildAnnoyIndex(conf, df) } } }