R1732 + W1514

This commit is contained in:
Jim O'Regan
2024-10-02 13:25:02 +00:00
parent c2569a1018
commit 0000f93021

View File

@@ -102,10 +102,8 @@ def main():
log.info("Dataloader loaded! Now computing stats...") log.info("Dataloader loaded! Now computing stats...")
params = compute_data_statistics(data_loader, cfg["n_feats"]) params = compute_data_statistics(data_loader, cfg["n_feats"])
print(params) print(params)
json.dump( with open(output_file, "w", encoding="utf-8") as dumpfile:
params, json.dump(params, dumpfile)
open(output_file, "w"),
)
if __name__ == "__main__": if __name__ == "__main__":