Close catalog before resetting it

NVIDIA · Sep 6, 2023 · d8a74ba · d8a74ba
1 parent f0ed106
commit d8a74ba
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/src/test/scala/com/nvidia/spark/rapids/RapidsHostMemoryStoreSuite.scala b/tests/src/test/scala/com/nvidia/spark/rapids/RapidsHostMemoryStoreSuite.scala
@@ -194,6 +194,7 @@ class RapidsHostMemoryStoreSuite extends AnyFunSuite with MockitoSugar {
   }
 
   test("get memory buffer after host spill") {
+    RapidsBufferCatalog.close()
     val sparkTypes = Array[DataType](IntegerType, StringType, DoubleType,
       DecimalType(ai.rapids.cudf.DType.DECIMAL64_MAX_PRECISION, 5))
     val spillPriority = -10