Format

apache · Jan 19, 2024 · 409869f · 409869f
1 parent 5f18f21
commit 409869f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/spark/src/main/scala/com/alibaba/graphar/writer/EdgeWriter.scala b/spark/src/main/scala/com/alibaba/graphar/writer/EdgeWriter.scala
@@ -151,8 +151,8 @@ object EdgeWriter {
       val offsetDfSchema = StructType(
         Seq(StructField(GeneralParams.offsetCol, IntegerType))
       )
-      val offsetDfArray: ParSeq[(Int, DataFrame)] = (0 until vertexChunkNum).par.map {
-        i =>
+      val offsetDfArray: ParSeq[(Int, DataFrame)] =
+        (0 until vertexChunkNum).par.map { i =>
           {
             val filterRDD = edgeCountsByPrimaryKey
               .filter(v => v._1 / vertexChunkSize == i)
@@ -179,7 +179,7 @@ object EdgeWriter {
             offsetChunk.persist(GeneralParams.defaultStorageLevel)
             (i, offsetChunk)
           }
-      }
+        }
       edgeCountsByPrimaryKey.unpersist() // unpersist the edgeCountsByPrimaryKey
       return (
         partitionEdgeDf,