Add annotation for mypy upgrade

DIRACGrid · Nov 16, 2023 · 548cb55 · 548cb55
1 parent b53eef5
commit 548cb55
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/diracx/db/sql/utils.py b/src/diracx/db/sql/utils.py
@@ -59,8 +59,8 @@ def substract_date(**kwargs: float) -> datetime:
     return datetime.now(tz=timezone.utc) - timedelta(**kwargs)
 
 
-Column = partial(RawColumn, nullable=False)
-NullColumn = partial(RawColumn, nullable=True)
+Column: partial[RawColumn] = partial(RawColumn, nullable=False)
+NullColumn: partial[RawColumn] = partial(RawColumn, nullable=True)
 DateNowColumn = partial(Column, DateTime(timezone=True), server_default=utcnow())