test(neuron): added a small script to prune test models

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()