Fix reading HDF5 from buffer objects

alubbock · Nov 11, 2018 · 1e905ac · 1e905ac
1 parent d8a5ea6
commit 1e905ac
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/thunor/io.py b/thunor/io.py
@@ -926,11 +926,14 @@ def _read_hdf_unstacked(filename_or_buffer):
     if isinstance(filename_or_buffer, str):
         hdf_kwargs['path'] = filename_or_buffer
     else:
+        if hasattr(filename_or_buffer, 'read') and \
+                callable(filename_or_buffer.read):
+            filename_or_buffer = filename_or_buffer.read()
         hdf_kwargs.update({
             'path': 'data.h5',
             'driver': 'H5FD_CORE',
             'driver_core_backing_store': 0,
-            'driver_core_image': filename_or_buffer.read()
+            'driver_core_image': filename_or_buffer
         })
     with pd.HDFStore(**hdf_kwargs) as hdf:
         df_assays = hdf['assays']