fix for tests

legend-exp · Oct 9, 2024 · 8e823c0 · 8e823c0
1 parent 2337326
commit 8e823c0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/pygama/evt/utils.py b/src/pygama/evt/utils.py
@@ -16,13 +16,19 @@
 H5DataLoc = namedtuple(
     "H5DataLoc", ("file", "group", "table_fmt"), defaults=3 * (None,)
 )
-DataInfo = namedtuple("DataInfo", ("raw"), defaults=1 * (None,))
+DataInfo = namedtuple("DataInfo", ("raw", "tcm", "evt"), defaults=3 * (None,))
 
 TCMData = namedtuple("TCMData", ("id", "idx", "cumulative_length"))
 
 
 def make_files_config(data: dict):
     if not isinstance(data, tuple):
+        if "raw" not in data:
+            data["raw"] = (None,)
+        if "tcm" not in data:
+            data["tcm"] = (None,)
+        if "evt" not in data:
+            data["evt"] = (None,)
         DataInfo = namedtuple(
             "DataInfo", tuple(data.keys()), defaults=len(data.keys()) * (None,)
         )