From 70cd8fa948b238e6f02b7febd899eabf84892936 Mon Sep 17 00:00:00 2001
From: Benjamin Bossan <benjamin.bossan@gmail.com>
Date: Fri, 5 Jan 2024 14:56:06 +0100
Subject: [PATCH] Fix failing clean_skops.py script

Return type from client.model_info.lastModified is now a tz-aware
datetime object.
---
 scripts/clean_skops.py | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/scripts/clean_skops.py b/scripts/clean_skops.py
index 31800bd7..c1945c5f 100644
--- a/scripts/clean_skops.py
+++ b/scripts/clean_skops.py
@@ -34,11 +34,9 @@
         # https://github.com/huggingface/moon-landing/issues/6034
         continue
 
-    age = (
-        datetime.datetime.now()
-        - datetime.datetime.fromisoformat(info.lastModified.rsplit(".", 1)[0])
-    ).days
-    if age < MAX_AGE:
+    now = datetime.datetime.now(info.lastModified.tzinfo)
+    age = now - info.lastModified
+    if age.days < MAX_AGE:
         print(f"Skipping model: {model_info.modelId}, age: {age}")
         continue
     print(f"deleting {model_info.modelId}, age: {age} days")
@@ -56,11 +54,9 @@
         # https://github.com/huggingface/moon-landing/issues/6034
         continue
 
-    age = (
-        datetime.datetime.now()
-        - datetime.datetime.fromisoformat(info.lastModified.rsplit(".", 1)[0])
-    ).days
-    if age < MAX_AGE:
+    now = datetime.datetime.now(info.lastModified.tzinfo)
+    age = now - info.lastModified
+    if age.days < MAX_AGE:
         print(f"Skipping space: {space_info.id}, age: {age}")
         continue
     print(f"deleting {space_info.id}, age: {age} days")