Don't try to delete from dict when key is not present.

pfizer-opensource · Jan 29, 2024 · 8aa183a · 8aa183a
1 parent aaccbc1
commit 8aa183a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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: