chore: pydantic 2

nextml-code · Feb 21, 2024 · 5361ba5 · 5361ba5
1 parent 70d1cbd
commit 5361ba5
Show file tree

Hide file tree

Showing 4 changed files with 1,261 additions and 1,002 deletions.
diff --git a/fast_s3/file.py b/fast_s3/file.py
@@ -13,17 +13,14 @@ class Status(str, Enum):
     error = "error"
 
 
-class File(BaseModel):
+class File(BaseModel, arbitrary_types_allowed=True):
     buffer: io.BytesIO
     future: TransferFuture
     path: Union[str, Path]
     status: Status = Status.pending
     exception: Optional[Exception] = None
 
-    class Config:
-        arbitrary_types_allowed = True
-
     def with_status(self, status: Status, exception: Optional[Exception] = None):
-        attributes = self.dict()
+        attributes = dict(self)
         attributes.update(status=status, exception=exception)
         return File(**attributes)
diff --git a/fast_s3/uploader.py b/fast_s3/uploader.py
@@ -13,13 +13,10 @@ class Status(str, Enum):
     error = "error"
 
 
-class Result(BaseModel):
+class Result(BaseModel, arbitrary_types_allowed=True):
     status: Status
     exception: Optional[Exception] = None
 
-    class Config:
-        arbitrary_types_allowed = True
-
 
 class Uploader:
     def __init__(