Skip to content

Commit b6ae529

Browse files
authored
use model_dump instead of dict (#8493)
1 parent 4173100 commit b6ae529

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/clients/test_lm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ def test_logprobs_included_when_requested():
301301
)
302302
result = lm("question")
303303
assert result[0]["text"] == "test answer"
304-
assert result[0]["logprobs"].dict() == {
304+
assert result[0]["logprobs"].model_dump() == {
305305
"content": [
306306
{
307307
"token": "test",

0 commit comments

Comments
 (0)