Skip to content

Commit f8f0460

Browse files
authored
Fix: remove tzinfo (#348)
1 parent ae407b8 commit f8f0460

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

src/libkernelbot/leaderboard_db.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,6 @@ def create_leaderboard(
8181
gpu_types: list | str,
8282
) -> int:
8383
# to prevent surprises, ensure we have specified a timezone
84-
assert deadline.tzinfo is not None
8584
try:
8685
task = definition.task
8786
self.cursor.execute(
@@ -260,17 +259,14 @@ def validate_identity(
260259
(identifier,),
261260
)
262261
row = self.cursor.fetchone()
263-
return {
264-
"user_id": row[0],
265-
"user_name": row[1],
266-
"id_type":id_type.value
267-
} if row else None
262+
return (
263+
{"user_id": row[0], "user_name": row[1], "id_type": id_type.value} if row else None
264+
)
268265
except psycopg2.Error as e:
269266
self.connection.rollback()
270267
logger.exception("Error validating %s %s", human_label, identifier, exc_info=e)
271268
raise KernelBotError(f"Error validating {human_label}") from e
272269

273-
274270
def create_submission(
275271
self,
276272
leaderboard: str,
@@ -390,7 +386,6 @@ def update_heartbeat_if_active(self, sub_id: int, ts: datetime.datetime) -> None
390386
logger.error("Failed to upsert submission job status. sub_id: '%s'", sub_id, exc_info=e)
391387
raise KernelBotError("Error updating job status") from e
392388

393-
394389
def upsert_submission_job_status(
395390
self,
396391
sub_id: int,

0 commit comments

Comments
 (0)