Skip to content

Commit

Permalink
Merge pull request #1086 from ikrommyd/fix-do-preprocess
Browse files Browse the repository at this point in the history
fix: output of `do_preprocess` saved in the wrong order of files
  • Loading branch information
lgray committed Apr 26, 2024
2 parents 35e96c2 + b248249 commit 8bc66c9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/coffea/dataset_tools/dataset_query.py
Expand Up @@ -556,11 +556,11 @@ def do_preprocess(
)
with gzip.open(f"{output_file}_available.json.gz", "wt") as file:
print(f"Saved available fileset chunks to {output_file}_available.json.gz")
json.dump(self.preprocessed_total, file, indent=2)
json.dump(self.preprocessed_available, file, indent=2)
with gzip.open(f"{output_file}_all.json.gz", "wt") as file:
print(f"Saved all fileset chunks to {output_file}_all.json.gz")
json.dump(self.preprocessed_available, file, indent=2)
return self.preprocessed_total, self.preprocessed_available
json.dump(self.preprocessed_total, file, indent=2)
return self.preprocessed_available, self.preprocessed_total

def load_dataset_definition(
self,
Expand Down

0 comments on commit 8bc66c9

Please sign in to comment.