diff --git a/src/main/scala/cognite/spark/v1/RawTableRelation.scala b/src/main/scala/cognite/spark/v1/RawTableRelation.scala index 313207568..85caf23ad 100644 --- a/src/main/scala/cognite/spark/v1/RawTableRelation.scala +++ b/src/main/scala/cognite/spark/v1/RawTableRelation.scala @@ -289,9 +289,9 @@ class RawTableRelation( config.maxOutstandingRawInsertRequests match { case Some(maxOutstandingRawInsertRequests) => Backpressure[IO](Backpressure.Strategy.Lossless, maxOutstandingRawInsertRequests) - .map { backpressure => - rows.grouped(batchSize).toVector.foreach { batch: Seq[Row] => - backpressure.metered(postRows(columnNames, batch)).unsafeRunSync() + .flatMap { backpressure => + rows.grouped(batchSize).toVector.traverse_ { batch: Seq[Row] => + backpressure.metered(postRows(columnNames, batch)) } } .unsafeRunSync()