Skip to content

Commit 1229a8b

Browse files
fix: remove unnecessary type ignore comments in Cerebras tests
1 parent ce53b2b commit 1229a8b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/models/test_openai.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3022,7 +3022,7 @@ async def test_openai_model_settings_temperature_ignored_on_gpt_5(allow_model_re
30223022

30233023

30243024
async def test_openai_model_cerebras_provider(allow_model_requests: None, cerebras_api_key: str):
3025-
m = OpenAIChatModel('llama3.3-70b', provider=CerebrasProvider(api_key=cerebras_api_key)) # type: ignore[arg-type]
3025+
m = OpenAIChatModel('llama3.3-70b', provider=CerebrasProvider(api_key=cerebras_api_key))
30263026
agent = Agent(m)
30273027

30283028
result = await agent.run('What is the capital of France?')
@@ -3034,15 +3034,15 @@ class Location(TypedDict):
30343034
city: str
30353035
country: str
30363036

3037-
m = OpenAIChatModel('qwen-3-coder-480b', provider=CerebrasProvider(api_key=cerebras_api_key)) # type: ignore[arg-type]
3037+
m = OpenAIChatModel('qwen-3-coder-480b', provider=CerebrasProvider(api_key=cerebras_api_key))
30383038
agent = Agent(m, output_type=Location)
30393039

30403040
result = await agent.run('What is the capital of France?')
30413041
assert result.output == snapshot({'city': 'Paris', 'country': 'France'})
30423042

30433043

30443044
async def test_openai_model_cerebras_provider_harmony(allow_model_requests: None, cerebras_api_key: str):
3045-
m = OpenAIChatModel('gpt-oss-120b', provider=CerebrasProvider(api_key=cerebras_api_key)) # type: ignore[arg-type]
3045+
m = OpenAIChatModel('gpt-oss-120b', provider=CerebrasProvider(api_key=cerebras_api_key))
30463046
agent = Agent(m)
30473047

30483048
result = await agent.run('What is the capital of France?')

0 commit comments

Comments
 (0)