bgc-setnet / source /scripts /manifest_external_benchmark.py
whiteh4t's picture
Release final BGC retrieval checkpoints and model card
c87881a verified
Raw
History Blame Contribute Delete
1.33 kB
#!/usr/bin/env python3
from __future__ import annotations
import argparse
import json
from bgc_retrieval.external_manifest import build_external_manifest
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--external-root", default="data/external")
parser.add_argument(
"--output", default="data/manifests/external_benchmark_manifest_v2.json"
)
parser.add_argument(
"--benchmark-commit", default="bb8500d60f90cb43397cc41de5aed396725aa800"
)
parser.add_argument(
"--archive-url",
default=(
"https://github.com/aswalker-lab/BGC-clustering-benchmark/archive/"
"bb8500d60f90cb43397cc41de5aed396725aa800.tar.gz"
),
)
parser.add_argument(
"--archive-sha256",
default="35d8484f00e4988d2e101038aa067fe1f93bb177157a1039aa457c0fc8deaf4f",
)
parser.add_argument(
"--supersedes", default="data/manifests/external_benchmark_manifest.json"
)
args = parser.parse_args()
manifest = build_external_manifest(
args.external_root,
args.output,
args.benchmark_commit,
args.archive_url,
args.archive_sha256,
args.supersedes,
)
print(json.dumps(manifest, indent=2, sort_keys=True))
if __name__ == "__main__":
main()