Skip to content

Commit e30ee69

Browse files
authored
Merge branch 'main' into apply-scalafmt
2 parents cecb52b + 576b5a6 commit e30ee69

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/main/scala/ldbc/snb/datagen/io/graphs.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ object graphs {
9292
SparkUI.job(getClass.getSimpleName, s"write $tpe") {
9393
val p = (sink.path / "graphs" / sink.format / PathComponent[GraphLike[M]].path(self) / tpe.entityPath).toString
9494
log.info(s"$tpe: Writing started")
95-
val opts = getFormatOptions(sink.format, self.mode)
95+
val opts = getFormatOptions(sink.format, self.mode, sink.formatOptions)
9696
the(dataset).write(DataFrameSink(p, sink.format, opts, SaveMode.Ignore))
9797
log.info(s"$tpe: Writing completed")
9898
}(dataset.sparkSession)
@@ -111,7 +111,7 @@ object graphs {
111111
import CacheFriendlyEntityOrdering._
112112

113113
override def write(self: Graph[M], sink: GraphSink): Unit = {
114-
val opts = getFormatOptions(sink.format, self.mode)
114+
val opts = getFormatOptions(sink.format, self.mode, sink.formatOptions)
115115
TreeMap(self.entities.mapValues(ev).toSeq: _*).foreach { case (tpe, BatchedEntity(snapshot, insertBatches, deleteBatches)) =>
116116
SparkUI.job(getClass.getSimpleName, s"write $tpe snapshot") {
117117
val p = (sink.path / "graphs" / sink.format / PathComponent[GraphLike[M]].path(self) / "initial_snapshot" / tpe.entityPath).toString

0 commit comments

Comments
 (0)