Update tools/hf_backup.py

tools/hf_backup.py

@@ -13,13 +13,16 @@ def list_backups(api: HfApi, repo_id: str, prefix: str):
     backs.sort()
     return backs
 
-def upload(args):
-    api = HfApi(token=args.token)
-    # 可选：确认数据集存在
+def ensure_dataset(api: HfApi, repo_id: str):
     try:
-        api.dataset_info(repo_id=args.repo)
+        api.dataset_info(repo_id=repo_id, use_auth_token=api.token)  # 兼容旧版
     except Exception:
-        pass
+        # 不存在则创建（可按需改为 public）
+        api.create_repo(repo_id=repo_id, repo_type="dataset", private=True, exist_ok=True)
+
+def upload(args):
+    api = HfApi(token=args.token)
+    ensure_dataset(api, args.repo)
 
     # 上传
     api.upload_file(
@@ -31,15 +34,15 @@ def upload(args):
 
     # 保留最新 N 份
     if args.max and args.max > 0:
-      backs = list_backups(api, args.repo, args.prefix)
-      if len(backs) > args.max:
-          to_del = backs[: len(backs) - args.max]
-          for f in to_del:
-              try:
-                  api.delete_file(path_in_repo=f, repo_id=args.repo, repo_type="dataset")
-              except Exception:
-                  # 删除失败不致命
-                  pass
+        backs = list_backups(api, args.repo, args.prefix)
+        if len(backs) > args.max:
+            to_del = backs[: len(backs) - args.max]
+            for f in to_del:
+                try:
+                    api.delete_file(path_in_repo=f, repo_id=args.repo, repo_type="dataset")
+                except Exception:
+                    # 删除失败不致命
+                    pass
 
 def restore(args):
     api = HfApi(token=args.token)