From bded9b4e16ede2166a40e2fb3679732217fb7064 Mon Sep 17 00:00:00 2001
From: Claire McGinty <clairem@spotify.com>
Date: Thu, 19 Sep 2024 15:06:11 -0400
Subject: [PATCH] simplify write

---
 jmh/src/test/scala/magnolify/jmh/MagnolifyBench.scala | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/jmh/src/test/scala/magnolify/jmh/MagnolifyBench.scala b/jmh/src/test/scala/magnolify/jmh/MagnolifyBench.scala
index 6e21046c..e3fd8335 100644
--- a/jmh/src/test/scala/magnolify/jmh/MagnolifyBench.scala
+++ b/jmh/src/test/scala/magnolify/jmh/MagnolifyBench.scala
@@ -244,9 +244,9 @@ object ParquetStates {
 
   @State(Scope.Benchmark)
   class WriteState[T](writeSupport: WriteSupport[T]) {
-    var writer: WriteSupport[T] = null
+    val writer = writeSupport
 
-    @Setup(Level.Iteration)
+    @Setup(Level.Trial)
     def setup(): Unit = {
       writeSupport.init(new PlainParquetConfiguration())
       // Use a no-op RecordConsumer; we want to measure only the record -> group conversion, and not pollute the
@@ -265,7 +265,6 @@ object ParquetStates {
         override def addFloat(value: Float): Unit = {}
         override def addDouble(value: Double): Unit = {}
       })
-      this.writer = writeSupport
     }
   }