write agg signatures to a file

binh-vu · Jan 22, 2024 · 5cd4304 · 5cd4304
1 parent 4bc59e2
commit 5cd4304
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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__":