load_data() returns the H5 file, wrapped in an np.array()

ArtificialIntelligenceToolkit · Sep 14, 2018 · c9809d3 · c9809d3
1 parent 2645283
commit c9809d3
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/conx/utils.py b/conx/utils.py
@@ -2095,17 +2095,14 @@ def shape(item, summary=False):
     else:
         return retval
 
-def load_data(filename, return_h5=False, *args, **kwargs):
+def load_data(filename, *args, **kwargs):
     """
     Load a numpy or h5 datafile.
     """
     import h5py
     if filename.endswith("h5"):
         h5 = h5py.File(filename, 'r')
-        if return_h5:
-            return h5
-        else:
-            return h5["data"][:]
+        return np.array(h5["data"])
     else:
         return np.load(filename, *args, **kwargs)