diff --git a/matcha/utils/generate_data_statistics.py b/matcha/utils/generate_data_statistics.py index 49ed3c1..305d806 100644 --- a/matcha/utils/generate_data_statistics.py +++ b/matcha/utils/generate_data_statistics.py @@ -102,10 +102,8 @@ def main(): log.info("Dataloader loaded! Now computing stats...") params = compute_data_statistics(data_loader, cfg["n_feats"]) print(params) - json.dump( - params, - open(output_file, "w"), - ) + with open(output_file, "w", encoding="utf-8") as dumpfile: + json.dump(params, dumpfile) if __name__ == "__main__":