From 5cd4304393f2d3155c61f9ad2bfab382c1f00750 Mon Sep 17 00:00:00 2001
From: Binh Vu <binh@toan2.com>
Date: Mon, 22 Jan 2024 02:08:54 +0000
Subject: [PATCH] write agg signatures to a file

---
 scripts/aggregate_datasets_signature.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/scripts/aggregate_datasets_signature.py b/scripts/aggregate_datasets_signature.py
index cc62f3f..3d4f542 100644
--- a/scripts/aggregate_datasets_signature.py
+++ b/scripts/aggregate_datasets_signature.py
@@ -12,7 +12,8 @@
 
 @click.command()
 @click.argument("input_dir", type=click.Path(exists=True))
-def make_signatures(input_dir: Path):
+def make_signatures(input_dir: Path | str):
+    input_dir = Path(input_dir)
     signatures = {}
     for sigfile in sorted(input_dir.glob("**/*/_SIGNATURE")):
         sig = DatasetSignature.from_dict(serde.json.deser(sigfile))
@@ -21,7 +22,8 @@ def make_signatures(input_dir: Path):
             "checksum": sig.checksum,
             "dependencies": sorted(sig.dependencies.keys()),
         }
-    return signatures
+
+    serde.json.ser(signatures, input_dir / "aggregated_signatures.json", indent=2)
 
 
 if __name__ == "__main__":