Merge pull request #120 from legend-exp/fix

Fix `read_as()` with file list of length 1

Author: gipert

src/lgdo/lh5/core.py

@@ -120,6 +120,7 @@ def read(
         lh5_obj = lh5_file[name]
     else:
         lh5_files = list(lh5_file)
+
         n_rows_read = 0
         obj_buf_is_new = False
 
@@ -353,5 +354,8 @@ def read_as(
     # NOTE: providing a buffer does not make much sense
     obj = read(name, lh5_file, **kwargs1)
 
+    if isinstance(obj, tuple):
+        obj = obj[0]
+
     # and finally return a view
     return obj.view_as(library, **kwargs2)

src/lgdo/types/vectorofvectors.py

@@ -503,7 +503,7 @@ def __iter__(self) -> Iterator[NDArray]:
     def __str__(self) -> str:
         string = self.view_as("ak").show(stream=None)
 
-        string = string.strip().removesuffix("]")
+        string = str(string).strip().removesuffix("]")
         string += "\n]"
 
         tmp_attrs = self.attrs.copy()

tests/lh5/test_core.py

@@ -38,6 +38,8 @@ def test_read_as(lh5_file):
     )
     assert obj1.equals(obj2)
 
+    obj2 = lh5.read_as("/data/struct/table", [lh5_file], "ak")
+
 
 def test_read_multiple_files(lh5_file):
     lh5_obj = lh5.read("/data/struct/array", [lh5_file, lh5_file, lh5_file])