From f4c72603de30e3038493f5004fd8c14016e91676 Mon Sep 17 00:00:00 2001
From: Roman Cattaneo <1116746+romanc@users.noreply.github.com>
Date: Tue, 27 May 2025 17:39:06 +0200
Subject: [PATCH] Fix warning in tests

---
 ndsl/halo/data_transformer.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ndsl/halo/data_transformer.py b/ndsl/halo/data_transformer.py
index 9f9ab2f6..f3133974 100644
--- a/ndsl/halo/data_transformer.py
+++ b/ndsl/halo/data_transformer.py
@@ -225,9 +225,12 @@ def finalize(self):
         self.synchronize()
 
         # Push the buffers back in the cache
-        Buffer.push_to_cache(self._pack_buffer)
+        if self._pack_buffer is not None:
+            Buffer.push_to_cache(self._pack_buffer)
         self._pack_buffer = None
-        Buffer.push_to_cache(self._unpack_buffer)
+
+        if self._unpack_buffer is not None:
+            Buffer.push_to_cache(self._unpack_buffer)
         self._unpack_buffer = None
 
     @staticmethod