From 8c1940445503fd6678d0961600f2be81622793a2 Mon Sep 17 00:00:00 2001 From: Max Nanis Date: Tue, 24 Feb 2026 17:26:15 -0500 Subject: Extensive use of type checking. Movement of pytest conf towards handling managers (for db agnostic unittest). Starting to organize pytests. --- jb/models/bonus.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'jb/models/bonus.py') diff --git a/jb/models/bonus.py b/jb/models/bonus.py index 564a32d..a536dd1 100644 --- a/jb/models/bonus.py +++ b/jb/models/bonus.py @@ -1,11 +1,10 @@ -from typing import Optional, Dict +from typing import Optional, Dict, Any from pydantic import BaseModel, Field, ConfigDict, PositiveInt from typing_extensions import Self -from jb.models.currency import USDCent +from generalresearchutils.currency import USDCent from jb.models.custom_types import AMTBoto3ID, AwareDatetimeISO, UUIDStr -from jb.models.definitions import PayoutStatus class Bonus(BaseModel): @@ -41,7 +40,7 @@ class Bonus(BaseModel): return d @classmethod - def from_postgres(cls, data: Dict) -> Self: + def from_postgres(cls, data: Dict[str, Any]) -> Self: data["amount"] = USDCent(round(data["amount"] * 100)) fields = set(cls.model_fields.keys()) data = {k: v for k, v in data.items() if k in fields} -- cgit v1.2.3