From 106d0ee39fb095a5894ac8a3de7502c6f0299581 Mon Sep 17 00:00:00 2001 From: S1ro1 Date: Sun, 31 Aug 2025 22:12:02 +0300 Subject: [PATCH] Fix: remove tzinfo --- src/libkernelbot/leaderboard_db.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/src/libkernelbot/leaderboard_db.py b/src/libkernelbot/leaderboard_db.py index d452c6bd..599a0d79 100644 --- a/src/libkernelbot/leaderboard_db.py +++ b/src/libkernelbot/leaderboard_db.py @@ -81,7 +81,6 @@ def create_leaderboard( gpu_types: list | str, ) -> int: # to prevent surprises, ensure we have specified a timezone - assert deadline.tzinfo is not None try: task = definition.task self.cursor.execute( @@ -260,17 +259,14 @@ def validate_identity( (identifier,), ) row = self.cursor.fetchone() - return { - "user_id": row[0], - "user_name": row[1], - "id_type":id_type.value - } if row else None + return ( + {"user_id": row[0], "user_name": row[1], "id_type": id_type.value} if row else None + ) except psycopg2.Error as e: self.connection.rollback() logger.exception("Error validating %s %s", human_label, identifier, exc_info=e) raise KernelBotError(f"Error validating {human_label}") from e - def create_submission( self, leaderboard: str, @@ -390,7 +386,6 @@ def update_heartbeat_if_active(self, sub_id: int, ts: datetime.datetime) -> None logger.error("Failed to upsert submission job status. sub_id: '%s'", sub_id, exc_info=e) raise KernelBotError("Error updating job status") from e - def upsert_submission_job_status( self, sub_id: int,