diff options
Diffstat (limited to 'jb/models/bonus.py')
| -rw-r--r-- | jb/models/bonus.py | 7 |
1 files changed, 3 insertions, 4 deletions
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} |
