From 8aa183a0d7c0f08a41e1d9b0df8db52522fa7784 Mon Sep 17 00:00:00 2001
From: Joren Retel <joren.retel@pfizer.com>
Date: Mon, 29 Jan 2024 16:17:54 +0100
Subject: [PATCH] Don't try to delete from dict when key is not present.

---
 bigwig_loader/collection.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/bigwig_loader/collection.py b/bigwig_loader/collection.py
index 45b48b2..1d89aab 100644
--- a/bigwig_loader/collection.py
+++ b/bigwig_loader/collection.py
@@ -80,8 +80,10 @@ def reset_gpu(self) -> None:
         """
 
         self._out = cp.zeros((len(self), 1, 1), dtype=cp.float32)
-        del self.__dict__["decoder"]
-        del self.__dict__["memory_bank"]
+        if "decoder" in self.__dict__:
+            del self.__dict__["decoder"]
+        if "memory_bank" in self.__dict__:
+            del self.__dict__["memory_bank"]
 
     @cached_property
     def decoder(self) -> Decoder: