2727__status__ = "Alpha"
2828
2929
30- from ..io_source import IoSource
31- from ..io_sources import IoSources
32- from modacor .dataclasses .messagehandler import *
33-
34- from os .path import abspath
3530from logging import WARNING
36- import numpy as np
31+ from os .path import abspath
32+
3733import h5py
34+ import numpy as np
35+
36+ from modacor .dataclasses .messagehandler import *
37+
38+ from ..io_source import IoSource
39+ from ..io_sources import IoSources
3840
3941
4042class HDFLoader (IoSource ):
@@ -46,22 +48,20 @@ def __init__(self, source_reference: str, logging_level = WARNING):
4648 self ._file_datasets = []
4749 self ._file_datasets_shapes = {}
4850
49-
50- def _open_file (self , file_path = None ):
51+ def _open_file (self , file_path = None ):
5152 if file_path is None :
52- error = ' No filepath given'
53+ error = " No filepath given"
5354 self .hdf_logger .log .error (error )
5455 raise OSError (error )
55-
56+
5657 try :
57- self ._file_reference = h5py .File (file_path , 'r' )
58+ self ._file_reference = h5py .File (file_path , "r" )
5859 self ._file_path = abspath (file_path )
5960 self ._file_reference .visititems (self ._find_datasets )
6061 except OSError as error :
6162 self .hdf_logger .logger .error (error )
6263 raise OSError (error )
6364
64-
6565 def _close_file (self ):
6666 try :
6767 self ._file_reference .close ()
@@ -73,9 +73,8 @@ def _close_file(self):
7373 self .hdf_logger .log .error (error )
7474 raise OSError (error )
7575
76-
7776 def _find_datasets (self , path_name , path_object ):
7877 """An internal function to be used to walk the tree of an HDF5 file and return a list of the datasets within"""
79- if ( isinstance (self ._file_reference [path_name ], h5py ._hl .dataset .Dataset ) ):
78+ if isinstance (self ._file_reference [path_name ], h5py ._hl .dataset .Dataset ):
8079 self ._file_datasets .append (path_name )
81- self ._file_datasets_shapes [path_name ] = self ._file_reference [path_name ].shape
80+ self ._file_datasets_shapes [path_name ] = self ._file_reference [path_name ].shape
0 commit comments