from huggingface_hub import hf_hub_download
import shutil
# cached_path = hf_hub_download(
#     repo_id="returnzeros/anyrobot_bimanual",
#     filename="ACT_DP_multitask"
# )
# shutil.copy(cached_path, "./policy")
# print(cached_path)


from huggingface_hub import list_repo_files, hf_hub_download
import os, shutil

repo_id = "returnzeros/anyrobot_bimanual"
local_root = "./anyrobot_bimanual_policy"

os.makedirs(local_root, exist_ok=True)

file_list = list_repo_files(repo_id)

for file in file_list:
    print(f"? Downloading: {file}")
    path = hf_hub_download(repo_id=repo_id, filename=file)
    
    local_path = os.path.join(local_root, file)
    os.makedirs(os.path.dirname(local_path), exist_ok=True)
    shutil.copy(path, local_path)

print(f"load root path : {local_root}")