From 6706c9b4d976fa45953040892ffb277b0c2430bf Mon Sep 17 00:00:00 2001
From: David Corvoysier <david@huggingface.co>
Date: Fri, 21 Feb 2025 15:02:52 +0000
Subject: [PATCH] test(neuron): added a small script to prune test models

---
 backends/neuron/tests/prune_test_models.py | 25 ++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 backends/neuron/tests/prune_test_models.py

diff --git a/backends/neuron/tests/prune_test_models.py b/backends/neuron/tests/prune_test_models.py
new file mode 100644
index 00000000..592b1070
--- /dev/null
+++ b/backends/neuron/tests/prune_test_models.py
@@ -0,0 +1,25 @@
+import os
+
+from argparse import ArgumentParser
+from huggingface_hub import HfApi
+
+
+def main():
+    parser = ArgumentParser()
+    parser.add_argument("--yes", action="store_true", default=False)
+    args = parser.parse_args()
+    api = HfApi()
+    models = api.list_models(search="optimum-internal-testing/neuron-tgi-testing")
+    for model in models:
+        if args.yes:
+            delete = True
+        else:
+            answer = input(f"Do you want to delete {model.id} [y/N] ?")
+            delete = (answer == "y")
+        if delete:
+            api.delete_repo(model.id)
+            print(f"Deleted {model.id}.")
+
+
+if __name__ == "__main__":
+    main()