Merge pull request #999 from PrefectHQ/fix-tests

`utcnow` is deprecated
PrefectHQ · Nov 30, 2024 · 9ae60e1 · 9ae60e1
2 parents 9d20165 + d5905d7
commit 9ae60e1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/cookbook/structured_outputs.py b/cookbook/structured_outputs.py
@@ -7,7 +7,7 @@
 import marvin
 from pydantic import AfterValidator, Field
 
-result = marvin.extract(
+result: list[str] = marvin.extract(
     data="nO, i HaVe NeVeR HeaRd oF uV",
     target=Annotated[
         str,

diff --git a/src/marvin/types.py b/src/marvin/types.py
@@ -283,7 +283,9 @@ class Image(MarvinType):
     data: Optional[bytes] = Field(default=None, repr=False)
     url: Optional[str] = None
     format: str = "png"
-    timestamp: datetime.datetime = Field(default_factory=datetime.datetime.utcnow)
+    timestamp: datetime.datetime = Field(
+        default_factory=lambda: datetime.datetime.now(datetime.timezone.utc)
+    )
     detail: Literal["auto", "low", "high"] = "auto"
 
     def __init__(self, data_or_url=None, **kwargs):
@@ -354,7 +356,9 @@ class Audio(MarvinType):
     data: bytes = Field(repr=False)
     _data_stream: Optional[AsyncIterator[bytes]] = PrivateAttr()
     url: Optional[Path] = None
-    timestamp: datetime.datetime = Field(default_factory=datetime.datetime.utcnow)
+    timestamp: datetime.datetime = Field(
+        default_factory=lambda: datetime.datetime.now(datetime.timezone.utc)
+    )
     format: Literal["wav", "mp3", "pcm"] = "pcm"
 
     def __init__(self, _data_stream: AsyncIterator[bytes] = None, **data):