From b4ad520229e50fc930e1252d30bd2f00ff9ff3b6 Mon Sep 17 00:00:00 2001
From: Felix <fzimmermann89@gmail.com>
Date: Fri, 21 Jul 2023 11:53:18 +0200
Subject: [PATCH] Make Dataset a Context Manager. Dataset can now be used in a
 with statement.

---
 ismrmrd/hdf5.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/ismrmrd/hdf5.py b/ismrmrd/hdf5.py
index 02ad91f..f391128 100644
--- a/ismrmrd/hdf5.py
+++ b/ismrmrd/hdf5.py
@@ -158,7 +158,13 @@ def __del__(self):
             self.close()
         except:
             pass
-        
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
     @property
     def _dataset(self):
         if self._dataset_name not in self._file:
@@ -167,7 +173,7 @@ def _dataset(self):
 
     def list(self):
         return self._dataset.keys()
-    
+
     def close(self):
         self._file.close()