From 259396e3cff399c0b0a3914e13ae1206d21b9acf Mon Sep 17 00:00:00 2001 From: adrinjalali Date: Fri, 2 Dec 2022 15:50:18 +0100 Subject: [PATCH] CI fix clean_skops.py --- scripts/clean_skops.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/scripts/clean_skops.py b/scripts/clean_skops.py index 68c287b5..01a14811 100644 --- a/scripts/clean_skops.py +++ b/scripts/clean_skops.py @@ -10,19 +10,19 @@ # This is the token for the skops user. TODO remove eventually, see issue #47 token = "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" -client = HfApi() -user = client.whoami(token=token)["name"] +client = HfApi(token=token) +user = client.whoami()["name"] answer = input( f"Are you sure you want to delete all repos under {user} older than 7 days? (y/[n])" ) if answer != "y": exit(1) -models = client.list_models(author=user) +models = [x for x in client.list_models(author=user)] print(f"Found {len(models)} models, checking their age...") for model_info in models: - info = client.model_info(model_info.modelId, token=token) + info = client.model_info(model_info.modelId) age = ( datetime.datetime.now() - datetime.datetime.fromisoformat(info.lastModified.rsplit(".", 1)[0]) @@ -31,4 +31,4 @@ print(f"Skipping model: {model_info.modelId}, age: {age}") continue print(f"deleting {model_info.modelId}, age: {age} days") - client.delete_repo(model_info.modelId, token=token) + client.delete_repo(model_info.modelId)