Skip to content

Commit 576b5a6

Browse files
authored
Merge pull request #341 from ldbc/fix-format-options
2 parents f96dfdb + 83216ca commit 576b5a6

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)
@@ -109,7 +109,7 @@ object graphs {
109109
import CacheFriendlyEntityOrdering._
110110

111111
override def write(self: Graph[M], sink: GraphSink): Unit = {
112-
val opts = getFormatOptions(sink.format, self.mode)
112+
val opts = getFormatOptions(sink.format, self.mode, sink.formatOptions)
113113
TreeMap(self.entities.mapValues(ev).toSeq: _*).foreach {
114114
case (tpe, BatchedEntity(snapshot, insertBatches, deleteBatches)) =>
115115
SparkUI.job(getClass.getSimpleName, s"write $tpe snapshot") {

0 commit comments

Comments
 (0)