allow passing pytables options on load

uchicago-cs · Nov 19, 2019 · 1e7a4df · 1e7a4df
1 parent 01af936
commit 1e7a4df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepdish/io/hdf5io.py b/deepdish/io/hdf5io.py
@@ -598,7 +598,7 @@ def save(path, data, compression='default'):
             group._v_attrs[DEEPDISH_IO_UNPACK] = True
 
 
-def load(path, group=None, sel=None, unpack=False):
+def load(path, group=None, sel=None, unpack=False, *args, **kwargs):
     """
     Loads an HDF5 saved with `save`.
 
@@ -632,7 +632,7 @@ def load(path, group=None, sel=None, unpack=False):
     save
 
     """
-    with tables.open_file(path, mode='r') as h5file:
+    with tables.open_file(path, mode='r', *args, **kwargs) as h5file:
         pathtable = {}  # dict to keep track of objects already loaded
         if group is not None:
             if isinstance(group, str):