#!/usr/bin/env python
from __future__ import annotations

import argparse

from assaylens.data.hf_upload import upload_artifact_folder


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("local_dir")
    parser.add_argument("--repo-id", default=None)
    args = parser.parse_args()
    repo_id = upload_artifact_folder(args.local_dir, args.repo_id)
    print(f"uploaded {args.local_dir} to dataset repo {repo_id}")


if __name__ == "__main__":
    main()