Skip to content

Commit 913c8be

Browse files
committed
Reset mock in test file
1 parent 802c7e5 commit 913c8be

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/llm/test_openai_llm.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
from neo4j_graphrag.llm.types import ToolCallResponse
2424

2525

26-
def get_mock_openai() -> Any:
27-
mock = cast(Any, MagicMock())
26+
def get_mock_openai() -> MagicMock:
27+
mock = MagicMock()
2828
mock.OpenAIError = openai.OpenAIError
2929
return mock
3030

0 commit comments

Comments
 (0)