From 41deff1ef14e35f77047bd9c2519174c5000a0c5 Mon Sep 17 00:00:00 2001
From: Felix <fzimmermann89@gmail.com>
Date: Fri, 6 Oct 2023 16:23:22 +0200
Subject: [PATCH] Make Dataset a Context Manager. (#68)

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()