Fix error message when successively loading the same dataset (#4422)

PennyLaneAI · Aug 1, 2023 · f91811d · f91811d
1 parent ff87866
commit f91811d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pennylane/data/base/dataset.py b/pennylane/data/base/dataset.py
@@ -226,10 +226,9 @@ def open(
         filepath = Path(filepath).expanduser()
 
         if mode == "copy":
-            f_to_copy = h5py.File(filepath, "r")
-            f = hdf5.create_group()
-            hdf5.copy_all(f_to_copy, f)
-            f_to_copy.close()
+            with h5py.File(filepath, "r") as f_to_copy:
+                f = hdf5.create_group()
+                hdf5.copy_all(f_to_copy, f)
         else:
             f = h5py.File(filepath, mode)
 

diff --git a/pennylane/data/data_manager/__init__.py b/pennylane/data/data_manager/__init__.py
@@ -211,7 +211,7 @@ def load(  # pylint: disable=too-many-arguments
             if result.exception() is not None:
                 raise result.exception()
 
-    return [Dataset.open(Path(dest_path), "r") for dest_path in dest_paths]
+    return [Dataset.open(Path(dest_path), "a") for dest_path in dest_paths]
 
 
 def list_datasets() -> dict: