InternalColumnarRDDConverterSuite was leaking

NVIDIA · Jan 13, 2023 · 7dbda44 · 7dbda44
1 parent c123170
commit 7dbda44
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.../test/scala/org/apache/spark/sql/rapids/execution/InternalColumnarRDDConverterSuite.scala b/.../test/scala/org/apache/spark/sql/rapids/execution/InternalColumnarRDDConverterSuite.scala
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021-2022, NVIDIA CORPORATION. All rights reserved.
+ * Copyright (c) 2021-2023, NVIDIA CORPORATION. All rights reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -300,6 +300,8 @@ class InternalColumnarRDDConverterSuite extends SparkQueryCompareTestSuite {
           case _: RapidsHostColumnVector => true
           case _ => false
         }
+        columns.foreach(_.close())
+        table.close()
         isRapidsHostColumnVector
       }).collect()
       assert(result.forall(_ == true))