Skip to content

Commit 6f45cae

Browse files
authored
Removing __future__ import annotations from message.py (#284)
1 parent 83eac90 commit 6f45cae

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/aviary/message.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
from __future__ import annotations
2-
31
import json
42
from collections.abc import Iterable
53
from typing import TYPE_CHECKING, ClassVar, Self
@@ -122,7 +120,7 @@ def create_message(
122120
cls,
123121
role: str = DEFAULT_ROLE,
124122
text: str | None = None,
125-
images: list[np.ndarray | str] | str | np.ndarray | None = None,
123+
images: "list[np.ndarray | str] | str | np.ndarray | None" = None,
126124
) -> Self:
127125
"""Create a message with optional multimodal (just images so far) support.
128126
@@ -173,7 +171,7 @@ class MalformedMessageError(ValueError):
173171
"""Error to throw if some aspect of a Message variant is malformed."""
174172

175173
@classmethod
176-
def common_retryable_errors_log_filter(cls, record: LogRecord) -> bool:
174+
def common_retryable_errors_log_filter(cls, record: "LogRecord") -> bool:
177175
"""
178176
Filter out common parsing failures not worth looking into from logs.
179177

0 commit comments

Comments
 (0)