fixup! Support Snowflake (spotify#5500)

turb · Nov 13, 2024 · b6c8906 · b6c8906
1 parent 8b0da46
commit b6c8906
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scio-snowflake/src/main/scala/com/spotify/scio/snowflake/SnowflakeIO.scala b/scio-snowflake/src/main/scala/com/spotify/scio/snowflake/SnowflakeIO.scala
@@ -50,7 +50,7 @@ object SnowflakeIO {
           .withRole(connectionOptions.role)
           .withWarehouse(connectionOptions.warehouse)
       )
-      .pipe(ds => connectionOptions.schema.map(schema => ds.withSchema(schema)).getOrElse(ds))
+      .pipe(ds => connectionOptions.schema.fold(ds)(ds.withSchema))
 
   private[snowflake] def buildCsvMapper[T](rowDecoder: RowDecoder[T]): CsvMapper[T] =
     new CsvMapper[T] {