fix dump_dataset in dump_processed.py (#116)

cms-btv-pog · Feb 11, 2025 · c5210af · c5210af
1 parent d6d4d00
commit c5210af
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/dump_processed.py b/scripts/dump_processed.py
@@ -50,12 +50,12 @@ def dump_dataset(output, fname, alljson):
         for o in old.keys():
             if o not in original_list.keys():
                 original_list[o] = []
-            original_list[o].append(old[o])
+            original_list[o].extend(old[o])
 
     for m in output.keys():
         for f in output[m].keys():
             if f not in list_from_coffea.keys():
-                list_from_coffea[f] = []
+                list_from_coffea[f] = list(output[m][f]["fname"])
             else:
                 list_from_coffea[f] += list(set(output[m][f]["fname"]))
     failed = {}
@@ -65,7 +65,7 @@ def dump_dataset(output, fname, alljson):
             failed[t] = original_list[t]
             continue
         for f in original_list[t]:
-            if not f in list_from_coffea[t]:
+            if f not in list_from_coffea[t]:
                 failed[t].append(f)
 
     with open(f"{fname}_failed_dataset.json", "w") as outfile: