Skip to content

Commit 57d051a

Browse files
committed
Add type ignore comments for protected method calls in tests
1 parent 2ea2a63 commit 57d051a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/models/test_bedrock.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1522,7 +1522,7 @@ async def test_cache_point_filtering():
15221522
from pydantic_ai.models.bedrock import BedrockConverseModel
15231523

15241524
# Test the static method directly
1525-
messages = await BedrockConverseModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()]), count())
1525+
messages = await BedrockConverseModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()]), count()) # pyright: ignore[reportPrivateUsage]
15261526
# CachePoint should be filtered out, message should still be valid
15271527
assert len(messages) == 1
15281528
assert messages[0]['role'] == 'user'

tests/models/test_huggingface.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1024,8 +1024,8 @@ async def test_cache_point_filtering():
10241024
from pydantic_ai.models.huggingface import HuggingFaceModel
10251025

10261026
# Test the static method directly
1027-
msg = await HuggingFaceModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()]))
1027+
msg = await HuggingFaceModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()])) # pyright: ignore[reportPrivateUsage]
10281028

10291029
# CachePoint should be filtered out
10301030
assert msg['role'] == 'user'
1031-
assert len(msg['content']) == 1
1031+
assert len(msg['content']) == 1 # pyright: ignore[reportUnknownArgumentType]

0 commit comments

Comments
 (0)