Skip to content

Commit c930698

Browse files
committed
Remove cast
1 parent 5fb122f commit c930698

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

tests/contrib/openai_agents/test_openai_tracing.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import datetime
22
import uuid
33
from datetime import timedelta
4-
from typing import Any, Optional, cast
4+
from typing import Any, Optional
55

66
from agents import Span, Trace, TracingProcessor
77
from agents.tracing import get_trace_provider
@@ -13,9 +13,6 @@
1313
TestModelProvider,
1414
set_open_ai_agent_temporal_overrides,
1515
)
16-
from temporalio.contrib.openai_agents._temporal_trace_provider import (
17-
TemporalTraceProvider,
18-
)
1916
from temporalio.contrib.pydantic import pydantic_data_converter
2017
from tests.contrib.openai_agents.test_openai import ResearchWorkflow, TestResearchModel
2118
from tests.helpers import new_worker
@@ -51,7 +48,7 @@ async def test_tracing(client: Client):
5148
client = Client(**new_config)
5249

5350
with set_open_ai_agent_temporal_overrides():
54-
provider = cast(TemporalTraceProvider, get_trace_provider())
51+
provider = get_trace_provider()
5552

5653
processor = MemoryTracingProcessor()
5754
provider.set_processors([processor])

0 commit comments

Comments
 (0)